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

Implement solver.save and solver.load; update README

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-01-23 12:57:15 -06:00
parent 480da41fa9
commit 5817f273e9
3 changed files with 62 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
miplearn/tests/test_solver.py
View File

@@ -14,3 +14,20 @@ def test_solver():
solver.solve(instance)
solver.fit()
solver.solve(instance)
def test_solve_save_load():
instance = KnapsackInstance2(weights=[23., 26., 20., 18.],
prices=[505., 352., 458., 220.],
capacity=67.)
solver = LearningSolver()
solver.solve(instance)
solver.fit()
solver.save("/tmp/knapsack_train.bin")
prev_x_train_len = len(solver.x_train)
prev_y_train_len = len(solver.y_train)
solver = LearningSolver()
solver.load("/tmp/knapsack_train.bin")
assert len(solver.x_train) == prev_x_train_len
assert len(solver.y_train) == prev_y_train_len