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

Convert VariableFeatures into dataclass

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-04 22:56:26 -05:00
parent 59f4f75a53
commit d79eec5da6
6 changed files with 97 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/components/primal.py
View File

@@ -136,7 +136,7 @@ class PrimalSolutionComponent(Component):
category_offset: Dict[Hashable, int] = {cat: 0 for cat in x.keys()}
for (var_name, var_dict) in features.variables.items():
for (idx, var_features) in var_dict.items():
category = var_features["Category"]
category = var_features.category
offset = category_offset[category]
category_offset[category] += 1
if y_pred[category][offset, 0]:
@@ -159,15 +159,15 @@ class PrimalSolutionComponent(Component):
solution = sample["Solution"]
for (var_name, var_dict) in features.variables.items():
for (idx, var_features) in var_dict.items():
category = var_features["Category"]
category = var_features.category
if category is None:
continue
if category not in x.keys():
x[category] = []
y[category] = []
f: List[float] = []
assert var_features["User features"] is not None
f += var_features["User features"]
assert var_features.user_features is not None
f += var_features.user_features
if "LP solution" in sample and sample["LP solution"] is not None:
lp_value = sample["LP solution"][var_name][idx]
if lp_value is not None: