From 59a1b9a41008dd987a0b4c3bb7e0f6bcea0ee135 Mon Sep 17 00:00:00 2001 From: "Alinson S. Xavier" Date: Fri, 28 Oct 2022 11:57:57 -0500 Subject: [PATCH] BB: Expose enable_plunging in collect --- src/bb/collect.jl | 2 ++ src/bb/varbranch/reliability.jl | 3 +++ 2 files changed, 5 insertions(+) diff --git a/src/bb/collect.jl b/src/bb/collect.jl index d63df18..40813d4 100644 --- a/src/bb/collect.jl +++ b/src/bb/collect.jl @@ -16,6 +16,7 @@ function collect!( gap_limit::Float64 = 1e-4, print_interval::Int = 5, branch_rule::VariableBranchingRule = ReliabilityBranching(collect = true), + enable_plunging = true, )::NodePool model = read_from_file(filename) mip = init(optimizer) @@ -36,6 +37,7 @@ function collect!( gap_limit, print_interval, branch_rule, + enable_plunging, ) h5 = Hdf5Sample(replace(filename, ".mps.gz" => ".h5")) diff --git a/src/bb/varbranch/reliability.jl b/src/bb/varbranch/reliability.jl index 5c45c81..e3bc29a 100644 --- a/src/bb/varbranch/reliability.jl +++ b/src/bb/varbranch/reliability.jl @@ -182,6 +182,9 @@ function find_branching_var( end function collect!(rule::ReliabilityBranching, h5) + if rule.stats.num_strong_branch_calls == 0 + return + end h5.put_array("bb_score_var_names", to_str_array(rule.stats.score_var_names)) h5.put_array("bb_score_features", vcat(rule.stats.score_features'...)) h5.put_array("bb_score_targets", rule.stats.score_targets)