ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 01:18:52 -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/dynamic_common.py
View File

@@ -52,7 +52,7 @@ class DynamicConstraintsComponent(Component):
cids: Dict[str, List[str]] = {}
constr_categories_dict = instance.get_constraint_categories()
constr_features_dict = instance.get_constraint_features()
instance_features = sample.get_vector("instance_features_user")
instance_features = sample.get_vector("instance_features")
assert instance_features is not None
for cid in self.known_cids:
# Initialize categories

2
miplearn/components/objective.py
View File

@@ -79,7 +79,7 @@ class ObjectiveValueComponent(Component):
) -> Tuple[Dict[str, List[List[float]]], Dict[str, List[List[float]]]]:
lp_instance_features = sample.get_vector("lp_instance_features")
if lp_instance_features is None:
lp_instance_features = sample.get_vector("instance_features_user")
lp_instance_features = sample.get_vector("instance_features")
assert lp_instance_features is not None
# Features

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")

4
miplearn/components/static_lazy.py
View File

@@ -204,14 +204,14 @@ class StaticLazyConstraintsComponent(Component):
x: Dict[str, List[List[float]]] = {}
y: Dict[str, List[List[float]]] = {}
cids: Dict[str, List[str]] = {}
instance_features = sample.get_vector("instance_features_user")
instance_features = sample.get_vector("instance_features")
constr_features = sample.get_vector_list("lp_constr_features")
constr_names = sample.get_vector("constr_names")
constr_categories = sample.get_vector("constr_categories")
constr_lazy = sample.get_vector("constr_lazy")
lazy_enforced = sample.get_set("lazy_enforced")
if constr_features is None:
constr_features = sample.get_vector_list("constr_features_user")
constr_features = sample.get_vector_list("constr_features")
assert instance_features is not None
assert constr_features is not None