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

Make compatible with MIPLearn 5b3a56f0; reformat source code

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-18 05:36:51 -05:00
parent 97b9fc9ad8
commit 39072a6290
18 changed files with 342 additions and 314 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
test/utils/benchmark_test.jl
View File

@@ -9,38 +9,24 @@ using DataFrames
@testset "BenchmarkRunner" begin
@info "Building training data..."
instances = [
build_knapsack_file_instance(),
build_knapsack_file_instance(),
]
stats = parallel_solve!(
LearningSolver(Cbc.Optimizer),
instances,
)
instances = [build_knapsack_file_instance(), build_knapsack_file_instance()]
stats = parallel_solve!(LearningSolver(Cbc.Optimizer), instances)
@test length(stats) == 2
@test stats[1] !== nothing
@test stats[2] !== nothing
benchmark = BenchmarkRunner(
solvers=Dict(
"baseline" => LearningSolver(
Cbc.Optimizer,
components=[],
),
"ml-exact" => LearningSolver(
Cbc.Optimizer,
),
"ml-heur" => LearningSolver(
Cbc.Optimizer,
mode="heuristic",
),
solvers = Dict(
"baseline" => LearningSolver(Cbc.Optimizer, components = []),
"ml-exact" => LearningSolver(Cbc.Optimizer),
"ml-heur" => LearningSolver(Cbc.Optimizer, mode = "heuristic"),
),
)
@info "Fitting..."
fit!(benchmark, instances)
@info "Benchmarking..."
parallel_solve!(benchmark, instances, n_trials=2)
parallel_solve!(benchmark, instances, n_trials = 2)
csv_filename = tempname()
write_csv!(benchmark, csv_filename)