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

Make extractor configurable

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-15 09:57:10 -05:00
parent 95e326f5f6
commit 39597287a6
2 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
miplearn/features.py
View File

@@ -132,23 +132,22 @@ class Sample:
class FeaturesExtractor:
def __init__(
self,
internal_solver: "InternalSolver",
with_sa: bool = True,
) -> None:
self.solver = internal_solver
self.with_sa = with_sa
def extract(
self,
instance: "Instance",
solver: "InternalSolver",
with_static: bool = True,
) -> Features:
features = Features()
features.variables = self.solver.get_variables(
features.variables = solver.get_variables(
with_static=with_static,
with_sa=self.with_sa,
)
features.constraints = self.solver.get_constraints(
features.constraints = solver.get_constraints(
with_static=with_static,
)
if with_static: