用 GW 算法求最大割问题

用 GW 算法求最大割问题

原文链接：https://www.zhihu.com/zvideo/1592226240470945792?utm_id=0

import cvxpy as cp
import numpy as np
from scipy.linalg import sqrtm

edges = [(0,1), (0,2), (1,3), (1,4), (2,3), (3,4)] # 图结构

X = cp.Variable((5,5), symmetric=True)
constraints = [X >> 0] # 半正定
constraints += [X[i,i] == 1 for i in range(5)] # 对角线为 1，因为我们需要单位向量
objective = sum(0.5*(1-X[i,j]) for (i,j) in edges) # 目标函数
prob = cp.Problem(cp.Maximize(objective), constraints)
prob.solve() # 正定矩阵

x = sqrtm(X.value) # 通过矩阵平方根来得到向量 x_i，这些向量与矩阵具有相同的维度
u = np.random.randn(5) # 随机超平面
x = np.sign(x@u) # 根据节点落在超平面的哪一侧为我们的节点选择标签

print(x) # 得到的结果 经分析可知 割数为 5

[ 1.+0.j -1.+0.j -1.+0.j  1.+0.j -1.+0.j]