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

Add variables to model features

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-03-30 21:29:33 -05:00
parent 1224613b1a
commit 9cf28f3cdc
3 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
miplearn/features.py
View File

@@ -21,10 +21,11 @@ class ModelFeaturesExtractor:
constraints: Dict[str, ConstraintFeatures] = {}
for cid in self.solver.get_constraint_ids():
constraints[cid] = {
"rhs": self.solver.get_constraint_rhs(cid),
"lhs": self.solver.get_constraint_lhs(cid),
"sense": self.solver.get_constraint_sense(cid),
"RHS": self.solver.get_constraint_rhs(cid),
"LHS": self.solver.get_constraint_lhs(cid),
"Sense": self.solver.get_constraint_sense(cid),
}
return {
"constraints": constraints,
"Constraints": constraints,
"Variables": self.solver.get_empty_solution(),
}

9
miplearn/types.py
View File

@@ -76,9 +76,9 @@ LearningSolveStats = TypedDict(
ConstraintFeatures = TypedDict(
"ConstraintFeatures",
{
"rhs": float,
"lhs": Dict[str, float],
"sense": str,
"RHS": float,
"LHS": Dict[str, float],
"Sense": str,
},
total=False,
)
@@ -86,7 +86,8 @@ ConstraintFeatures = TypedDict(
ModelFeatures = TypedDict(
"ModelFeatures",
{
"constraints": Dict[str, ConstraintFeatures],
"Variables": Solution,
"Constraints": Dict[str, ConstraintFeatures],
},
total=False,
)