ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 09:28:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

STAB: Switch to clique formulation

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-04-15 19:42:21 -05:00
parent 9c3de5d7ce
commit 8f3f880afd
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/python/miplearn/problems/stab.py
View File

@@ -101,14 +101,13 @@ class MaxWeightStableSetInstance(Instance):
def to_model(self): def to_model(self):
nodes = list(self.graph.nodes) nodes = list(self.graph.nodes)
edges = list(self.graph.edges)
self.model = model = pe.ConcreteModel() self.model = model = pe.ConcreteModel()
model.x = pe.Var(nodes, domain=pe.Binary) model.x = pe.Var(nodes, domain=pe.Binary)
model.OBJ = pe.Objective(rule=lambda m : sum(m.x[v] * self.weights[v] for v in nodes), model.OBJ = pe.Objective(rule=lambda m : sum(m.x[v] * self.weights[v] for v in nodes),
sense=pe.maximize) sense=pe.maximize)
model.edge_eqs = pe.ConstraintList() model.clique_eqs = pe.ConstraintList()
for edge in edges: for clique in nx.find_cliques(self.graph):
model.edge_eqs.add(model.x[edge[0]] + model.x[edge[1]] <= 1) model.clique_eqs.add(sum(model.x[i] for i in clique) <= 1)
return model return model