ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -06:00
Code Issues Packages Projects Releases Wiki Activity

Make InternalSolver clonable

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-07 19:52:21 -05:00
parent ebccde6a03
commit 0232219a0e
12 changed files with 69 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
benchmark/benchmark.py
View File

@@ -62,7 +62,7 @@ def train(args):
PickleGzInstance(f) for f in glob.glob(f"{basepath}/train/*.gz")
]
solver = LearningSolver(
solver=lambda: GurobiPyomoSolver(
solver=GurobiPyomoSolver(
params={
"TimeLimit": int(args["--train-time-limit"]),
"Threads": int(args["--solver-threads"]),
@@ -83,7 +83,7 @@ def test_baseline(args):
if not os.path.isfile(csv_filename):
solvers = {
"baseline": LearningSolver(
solver=lambda: GurobiPyomoSolver(
solver=GurobiPyomoSolver(
params={
"TimeLimit": int(args["--test-time-limit"]),
"Threads": int(args["--solver-threads"]),
@@ -107,7 +107,7 @@ def test_ml(args):
if not os.path.isfile(csv_filename):
solvers = {
"ml-exact": LearningSolver(
solver=lambda: GurobiPyomoSolver(
solver=GurobiPyomoSolver(
params={
"TimeLimit": int(args["--test-time-limit"]),
"Threads": int(args["--solver-threads"]),
@@ -115,7 +115,7 @@ def test_ml(args):
),
),
"ml-heuristic": LearningSolver(
solver=lambda: GurobiPyomoSolver(
solver=GurobiPyomoSolver(
params={
"TimeLimit": int(args["--test-time-limit"]),
"Threads": int(args["--solver-threads"]),