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

Benchmark: Add extra columns to CSV

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-01-12 11:22:42 -06:00
parent f77d1d5de9
commit c9ad7a3f56
5 changed files with 60 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/tests/test_benchmark.py
View File

@@ -27,11 +27,11 @@ def test_benchmark():
benchmark = BenchmarkRunner(test_solvers)
benchmark.fit(train_instances)
benchmark.parallel_solve(test_instances, n_jobs=2, n_trials=2)
assert benchmark.raw_results().values.shape == (12, 16)
assert benchmark.raw_results().values.shape == (12, 18)
benchmark.save_results("/tmp/benchmark.csv")
assert os.path.isfile("/tmp/benchmark.csv")
benchmark = BenchmarkRunner(test_solvers)
benchmark.load_results("/tmp/benchmark.csv")
assert benchmark.raw_results().values.shape == (12, 16)
assert benchmark.raw_results().values.shape == (12, 18)