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

Fix solve_lp_first=False and add tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-21 08:25:57 -06:00
parent 06402516e6
commit a1b959755c
2 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
miplearn/solvers/learning.py
View File

@@ -225,6 +225,9 @@ class LearningSolver:
training_sample["LP solution"] = self.internal_solver.get_solution() training_sample["LP solution"] = self.internal_solver.get_solution()
training_sample["LP value"] = stats["Optimal value"] training_sample["LP value"] = stats["Optimal value"]
training_sample["LP log"] = stats["Log"] training_sample["LP log"] = stats["Log"]
else:
training_sample["LP solution"] = self.internal_solver.get_empty_solution()
training_sample["LP value"] = 0
# Before-solve callbacks # Before-solve callbacks
logger.debug("Running before_solve callbacks...") logger.debug("Running before_solve callbacks...")

13
miplearn/solvers/tests/test_learning_solver.py
View File

@@ -47,6 +47,19 @@ def test_learning_solver():
pickle.dump(solver, file) pickle.dump(solver, file)
def test_solve_without_lp():
for internal_solver in _get_internal_solvers():
logger.info("Solver: %s" % internal_solver)
instance = _get_instance(internal_solver)
solver = LearningSolver(
solver=internal_solver,
solve_lp_first=False,
)
solver.solve(instance)
solver.fit([instance])
solver.solve(instance)
def test_parallel_solve(): def test_parallel_solve():
for internal_solver in _get_internal_solvers(): for internal_solver in _get_internal_solvers():
instances = [_get_instance(internal_solver) for _ in range(10)] instances = [_get_instance(internal_solver) for _ in range(10)]