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

Add node_limit option to LearningSolver

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-04-24 07:50:42 -05:00
parent 696d66eba9
commit bdecde47fe
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/python/miplearn/solvers/learning.py
View File

@@ -48,8 +48,8 @@ class LearningSolver:
mode="exact", mode="exact",
solver="gurobi", solver="gurobi",
threads=4, threads=4,
time_limit=None): time_limit=None,
node_limit=None):
self.components = {} self.components = {}
self.mode = mode self.mode = mode
self.internal_solver = None self.internal_solver = None
@@ -58,6 +58,7 @@ class LearningSolver:
self.time_limit = time_limit self.time_limit = time_limit
self.gap_tolerance = gap_tolerance self.gap_tolerance = gap_tolerance
self.tee = False self.tee = False
self.node_limit = node_limit
if components is not None: if components is not None:
for comp in components: for comp in components:
@@ -86,6 +87,8 @@ class LearningSolver:
solver.set_time_limit(self.time_limit) solver.set_time_limit(self.time_limit)
if self.gap_tolerance is not None: if self.gap_tolerance is not None:
solver.set_gap_tolerance(self.gap_tolerance) solver.set_gap_tolerance(self.gap_tolerance)
if self.node_limit is not None:
solver.set_node_limit(self.node_limit)
return solver return solver
def solve(self, def solve(self,