Allenfenqu/objective_func.py

14 lines
485 B
Python
Raw Normal View History

2024-03-20 12:25:06 +08:00
from compute_cut_vol1 import compute_cut_vol
import numpy as np
def objective_func(G, partition,yanlinks,result_array):
"""
计算目标函数的值
G: networkx
partition: 分区的结果是一个列表列表中的每个元素是一个集合表示一个分区
返回值: 目标函数的值
"""
cut_value, vol_value = compute_cut_vol(G, partition,yanlinks,result_array)
return vol_value#cut_value / vol_value if vol_value != 0 else np.inf