ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-07 18:08: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

15
miplearn/benchmark.py
View File

@@ -13,11 +13,6 @@ class BenchmarkRunner:
self.solvers = solvers
self.results = None
def load_fit(self, filename):
for (name, solver) in self.solvers.items():
solver.load(filename)
solver.fit()
def parallel_solve(self, instances, n_jobs=1, n_trials=1):
if self.results is None:
self.results = pd.DataFrame(columns=["Solver",
@@ -78,4 +73,12 @@ class BenchmarkRunner:
self.results.to_csv(filename)
def load_results(self, filename):
self.results = pd.read_csv(filename, index_col=0)
self.results = pd.read_csv(filename, index_col=0)
def load_state(self, filename):
for (name, solver) in self.solvers.items():
solver.load_state(filename)
def fit(self):
for (name, solver) in self.solvers.items():
solver.fit()