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

Modularize LearningSolver into components; implement branch-priority

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-01-28 13:35:51 -06:00
parent 897743fce7
commit 6a29411df3
11 changed files with 348 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
miplearn/tests/test_benchmark.py
View File

@@ -19,15 +19,16 @@ def test_benchmark():
# Training phase...
training_solver = LearningSolver()
training_solver.parallel_solve(train_instances, n_jobs=10)
training_solver.save("data.bin")
training_solver.fit()
training_solver.save_state("data.bin")
# Test phase...
test_solvers = {
"Strategy A": LearningSolver(ws_predictor=None),
"Strategy B": LearningSolver(ws_predictor=None),
"Strategy A": LearningSolver(),
"Strategy B": LearningSolver(),
}
benchmark = BenchmarkRunner(test_solvers)
benchmark.load_fit("data.bin")
benchmark.load_state("data.bin")
benchmark.parallel_solve(test_instances, n_jobs=2, n_trials=2)
assert benchmark.raw_results().values.shape == (12,12)