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

Minor changes

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-01-29 08:36:36 -06:00
parent 5f2e81322c
commit b48dfad3aa
5 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/warmstart.py
View File

@@ -50,7 +50,7 @@ class WarmStartPredictor(ABC):
class LogisticWarmStartPredictor(WarmStartPredictor):
def __init__(self,
min_samples=100,
thr_fix=[0.99, 0.99],
thr_fix=[0.95, 0.95],
thr_balance=[0.95, 0.95],
thr_score=[0.95, 0.95]):
super().__init__()
@@ -151,8 +151,12 @@ class WarmStartComponent(Component):
if self.mode == "heuristic":
if ws[i,0] == 1:
var[index].fix(0)
if solver.is_persistent:
solver.internal_solver.update_var(var[index])
elif ws[i,1] == 1:
var[index].fix(1)
if solver.is_persistent:
solver.internal_solver.update_var(var[index])
else:
if ws[i,0] == 1:
var[index].value = 0