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

Only relax binary variables; do not clear value of fixed variables

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-03-01 11:13:17 -06:00
parent c4faa0c708
commit 570ac09286
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/solvers.py
View File

@@ -44,14 +44,15 @@ class InternalSolver:
self.solver.set_instance(self.model)
# Relax domain
from pyomo.core.base.set_types import Reals
from pyomo.core.base.set_types import Reals, Binary
original_domains = []
for (idx, var) in enumerate(self.model.component_data_objects(Var)):
original_domains += [var.domain]
lb, ub = var.bounds
if var.domain == Binary:
var.domain = Reals
var.setlb(lb)
var.setub(ub)
var.domain = Reals
self.solver.update_var(var)
# Solve LP relaxation
@@ -59,6 +60,7 @@ class InternalSolver:
# Restore domains
for (idx, var) in enumerate(self.model.component_data_objects(Var)):
if original_domains[idx] == Binary:
var.domain = original_domains[idx]
self.solver.update_var(var)
@@ -69,6 +71,8 @@ class InternalSolver:
def clear_values(self):
for var in self.model.component_objects(Var):
for index in var:
if var[index].fixed:
continue
var[index].value = None
def get_solution(self):