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

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(),
}