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

Primal: In evaluate, skip variables without predictions

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-10-05 11:28:30 -05:00
parent 0cf88ded89
commit 4144b8b6b3
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
miplearn/components/primal.py
View File

@@ -119,6 +119,8 @@ class PrimalSolutionComponent(Component):
vars_all, vars_one, vars_zero = set(), set(), set() vars_all, vars_one, vars_zero = set(), set(), set()
pred_one_positive, pred_zero_positive = set(), set() pred_one_positive, pred_zero_positive = set(), set()
for (varname, var_dict) in solution_actual.items(): for (varname, var_dict) in solution_actual.items():
if varname not in solution_pred.keys():
continue
for (idx, value) in var_dict.items(): for (idx, value) in var_dict.items():
vars_all.add((varname, idx)) vars_all.add((varname, idx))
if value > 0.5: if value > 0.5: