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

Make LearningSolver work with persistent solvers

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-01-22 15:20:59 -06:00
parent f538356bf6
commit e13eba86b0
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
miplearn/solvers.py
View File

@@ -57,7 +57,7 @@ class LearningSolver:
var[index].value = 0 var[index].value = 0
# Solve MILP # Solve MILP
self.parent_solver.solve(model, tee=tee, warmstart=True) self._solve(model, tee=tee)
# Update y_train # Update y_train
for category in var_split.keys(): for category in var_split.keys():
@@ -78,5 +78,9 @@ class LearningSolver:
self.ws_predictors[category] = WarmStartPredictor() self.ws_predictors[category] = WarmStartPredictor()
self.ws_predictors[category].fit(x_train, y_train) self.ws_predictors[category].fit(x_train, y_train)
def _solve(self, tee): def _solve(self, model, tee=False):
self.parent_solver.solve(tee=tee) if hasattr(self.parent_solver, "set_instance"):
self.parent_solver.set_instance(model)
self.parent_solver.solve(tee=tee, warmstart=True)
else:
self.parent_solver.solve(model, tee=tee, warmstart=True)