Allenfenqu/objective_func.py

14 lines
485 B
Python

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