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

Don't include intermediary features in sample; rename some keys

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-08-04 13:22:12 -05:00
parent 865a4b2f40
commit 10eed9b306
10 changed files with 33 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
miplearn/components/primal.py
View File

@@ -142,7 +142,7 @@ class PrimalSolutionComponent(Component):
) -> Tuple[Dict[Category, List[List[float]]], Dict[Category, List[List[float]]]]:
x: Dict = {}
y: Dict = {}
instance_features = sample.get_vector("instance_features_user")
instance_features = sample.get_vector("instance_features")
mip_var_values = sample.get_vector("mip_var_values")
var_features = sample.get_vector_list("lp_var_features")
var_names = sample.get_vector("var_names")