Add customizable branch priority; add more metrics to BenchmarkRunner

2025-12-07 18:08:51 -06:00 · 2020-01-28 06:51:49 -06:00
parent 99ac7aa718
commit f7d20ed52b
4 changed files with 70 additions and 26 deletions
--- a/miplearn/tests/test_solver.py
+++ b/miplearn/tests/test_solver.py
@@ -40,4 +40,11 @@ def test_parallel_solve():
    solver = LearningSolver()
    solver.parallel_solve(instances, n_jobs=3)
    assert len(solver.x_train[0]) == 10
-    assert len(solver.y_train[0]) == 10
+    assert len(solver.y_train[0]) == 10
+    
+def test_solver_random_branch_priority():
+    instance = KnapsackInstance2(weights=[23., 26., 20., 18.],
+                                 prices=[505., 352., 458., 220.],
+                                 capacity=67.)
+    solver = LearningSolver(branch_priority=[1, 2, 3, 4])
+    solver.solve(instance, tee=True)