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

ConstraintFeatures: Store lhs and sense

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-03-02 18:14:07 -06:00
parent 3a60deac63
commit b6ea0c5f1b
6 changed files with 61 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
miplearn/features.py
View File

@@ -2,9 +2,9 @@
# Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
# Released under the modified BSD license. See COPYING.md for more details.
from typing import TYPE_CHECKING
from typing import TYPE_CHECKING, Dict
from miplearn.types import ModelFeatures
from miplearn.types import ModelFeatures, ConstraintFeatures
if TYPE_CHECKING:
from miplearn import InternalSolver
@@ -15,12 +15,16 @@ class ModelFeaturesExtractor:
self,
internal_solver: "InternalSolver",
) -> None:
self.internal_solver = internal_solver
self.solver = internal_solver
def extract(self) -> ModelFeatures:
rhs = {}
for cid in self.internal_solver.get_constraint_ids():
rhs[cid] = self.internal_solver.get_constraint_rhs(cid)
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),
}
return {
"ConstraintRHS": rhs,
"constraints": constraints,
}