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 AdaptiveSolver; reorganize imports

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-02-21 11:22:44 -06:00
parent 1213f1d0a5
commit d3b5b43b94
4 changed files with 261 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/components/tests/test_warmstart.py
View File

@@ -26,12 +26,16 @@ def test_warm_start_save_load():
comp = solver.components["warm-start"]
assert comp.x_train["default"].shape == (8, 6)
assert comp.y_train["default"].shape == (8, 2)
assert "default" in comp.predictors.keys()
assert ("default", 0) in comp.predictors.keys()
assert ("default", 1) in comp.predictors.keys()
solver.save_state(state_file.name)
solver.solve(_get_instances()[0])
solver = LearningSolver(components={"warm-start": WarmStartComponent()})
solver.load_state(state_file.name)
comp = solver.components["warm-start"]
assert comp.x_train["default"].shape == (8, 6)
assert comp.y_train["default"].shape == (8, 2)
assert "default" in comp.predictors.keys()
assert ("default", 0) in comp.predictors.keys()
assert ("default", 1) in comp.predictors.keys()