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

Rename features.constraints to constraints_old

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-15 11:00:52 -05:00
parent 8f73d87d2d
commit 0e9c8b0a49
4 changed files with 23 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
miplearn/components/static_lazy.py
View File

@@ -78,7 +78,7 @@ class StaticLazyConstraintsComponent(Component):
assert solver.internal_solver is not None
assert sample.after_load is not None
assert sample.after_load.instance is not None
assert sample.after_load.constraints is not None
assert sample.after_load.constraints_old is not None
logger.info("Predicting violated (static) lazy constraints...")
if sample.after_load.instance.lazy_constraint_count == 0:
@@ -86,7 +86,7 @@ class StaticLazyConstraintsComponent(Component):
self.enforced_cids = set(self.sample_predict(sample))
logger.info("Moving lazy constraints to the pool...")
self.pool = {}
for (cid, cdict) in sample.after_load.constraints.items():
for (cid, cdict) in sample.after_load.constraints_old.items():
if cdict.lazy and cid not in self.enforced_cids:
self.pool[cid] = cdict
solver.internal_solver.remove_constraint(cid)
@@ -194,8 +194,8 @@ class StaticLazyConstraintsComponent(Component):
y: Dict[Hashable, List[List[float]]] = {}
cids: Dict[Hashable, List[str]] = {}
assert sample.after_load is not None
assert sample.after_load.constraints is not None
for (cid, constr) in sample.after_load.constraints.items():
assert sample.after_load.constraints_old is not None
for (cid, constr) in sample.after_load.constraints_old.items():
# Initialize categories
if not constr.lazy:
continue
@@ -213,9 +213,9 @@ class StaticLazyConstraintsComponent(Component):
sf = sample.after_lp
assert sf.instance is not None
features = list(sf.instance.to_list())
assert sf.constraints is not None
assert sf.constraints[cid] is not None
features.extend(sf.constraints[cid].to_list())
assert sf.constraints_old is not None
assert sf.constraints_old[cid] is not None
features.extend(sf.constraints_old[cid].to_list())
x[category].append(features)
cids[category].append(cid)

12
miplearn/features.py
View File

@@ -116,7 +116,7 @@ class Constraint:
class Features:
instance: Optional[InstanceFeatures] = None
variables: Optional[VariableFeatures] = None
constraints: Optional[Dict[str, Constraint]] = None
constraints_old: Optional[Dict[str, Constraint]] = None
lp_solve: Optional["LPSolveStats"] = None
mip_solve: Optional["MIPSolveStats"] = None
extra: Optional[Dict] = None
@@ -147,7 +147,7 @@ class FeaturesExtractor:
with_static=with_static,
with_sa=self.with_sa,
)
features.constraints = solver.get_constraints(
features.constraints_old = solver.get_constraints(
with_static=with_static,
)
if with_static:
@@ -201,9 +201,9 @@ class FeaturesExtractor:
instance: "Instance",
features: Features,
) -> None:
assert features.constraints is not None
assert features.constraints_old is not None
has_static_lazy = instance.has_static_lazy_constraints()
for (cid, constr) in features.constraints.items():
for (cid, constr) in features.constraints_old.items():
user_features = None
category = instance.get_constraint_category(cid)
if category is not None:
@@ -232,7 +232,7 @@ class FeaturesExtractor:
instance: "Instance",
features: Features,
) -> None:
assert features.constraints is not None
assert features.constraints_old is not None
user_features = instance.get_instance_features()
if isinstance(user_features, np.ndarray):
user_features = user_features.tolist()
@@ -246,7 +246,7 @@ class FeaturesExtractor:
f"Found {type(v).__name__} instead."
)
lazy_count = 0
for (cid, cdict) in features.constraints.items():
for (cid, cdict) in features.constraints_old.items():
if cdict.lazy:
lazy_count += 1
features.instance = InstanceFeatures(