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

Update benchmark scripts

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-02-22 22:56:53 -06:00
parent 9e4004ae8a
commit 15ee7f51ee
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
benchmark/benchmark.py
View File

@@ -30,6 +30,7 @@ logging.getLogger('pyomo.core').setLevel(logging.ERROR)
n_jobs = 10
time_limit = 300
internal_solver = "gurobi"
args = docopt(__doc__)
basepath = args["<challenge>"]
@@ -52,9 +53,11 @@ def train():
problem_name, challenge_name = args["<challenge>"].split("/")
pkg = importlib.import_module("miplearn.problems.%s" % problem_name)
challenge = getattr(pkg, challenge_name)()
train_instances = challenge.training_instances[:10]
test_instances = challenge.test_instances[:10]
solver = LearningSolver(time_limit=time_limit, components={})
train_instances = challenge.training_instances
test_instances = challenge.test_instances
solver = LearningSolver(time_limit=time_limit,
solver=internal_solver,
components={})
solver.parallel_solve(train_instances, n_jobs=n_jobs)
solver.fit(n_jobs=n_jobs)
save(train_instances, "%s/train_instances.bin" % basepath)