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

BB: Expose enable_plunging in collect

Browse Source
This commit is contained in:
Alinson S. Xavier
2022-10-28 11:57:57 -05:00
parent 38c4e41720
commit 59a1b9a410
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/bb/collect.jl
View File

@@ -16,6 +16,7 @@ function collect!(
gap_limit::Float64 = 1e-4, gap_limit::Float64 = 1e-4,
print_interval::Int = 5, print_interval::Int = 5,
branch_rule::VariableBranchingRule = ReliabilityBranching(collect = true), branch_rule::VariableBranchingRule = ReliabilityBranching(collect = true),
enable_plunging = true,
)::NodePool )::NodePool
model = read_from_file(filename) model = read_from_file(filename)
mip = init(optimizer) mip = init(optimizer)
@@ -36,6 +37,7 @@ function collect!(
gap_limit, gap_limit,
print_interval, print_interval,
branch_rule, branch_rule,
enable_plunging,
) )
h5 = Hdf5Sample(replace(filename, ".mps.gz" => ".h5")) h5 = Hdf5Sample(replace(filename, ".mps.gz" => ".h5"))

3
src/bb/varbranch/reliability.jl
View File

@@ -182,6 +182,9 @@ function find_branching_var(
end end
function collect!(rule::ReliabilityBranching, h5) 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_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_features", vcat(rule.stats.score_features'...))
h5.put_array("bb_score_targets", rule.stats.score_targets) h5.put_array("bb_score_targets", rule.stats.score_targets)