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

Extract more features to ConstraintFeatures

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-15 12:21:19 -05:00
parent 230d13a5c0
commit 18521331c9
5 changed files with 96 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
miplearn/features.py
View File

@@ -148,6 +148,7 @@ class Constraint:
class Features:
instance: Optional[InstanceFeatures] = None
variables: Optional[VariableFeatures] = None
constraints: Optional[ConstraintFeatures] = None
constraints_old: Optional[Dict[str, Constraint]] = None
lp_solve: Optional["LPSolveStats"] = None
mip_solve: Optional["MIPSolveStats"] = None
@@ -179,6 +180,10 @@ class FeaturesExtractor:
with_static=with_static,
with_sa=self.with_sa,
)
features.constraints = solver.get_constraints(
with_static=with_static,
with_sa=self.with_sa,
)
features.constraints_old = solver.get_constraints_old(
with_static=with_static,
)