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

LearningSolver: add extract_sa, extract_lhs arguments

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-06-28 17:34:15 -05:00
parent 101bd94a5b
commit 5b4b8adee5
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
miplearn/solvers/learning.py
View File

@@ -101,11 +101,16 @@ class LearningSolver:
solve_lp: bool = True, solve_lp: bool = True,
simulate_perfect: bool = False, simulate_perfect: bool = False,
extractor: Optional[FeaturesExtractor] = None, extractor: Optional[FeaturesExtractor] = None,
extract_lhs: bool = True,
extract_sa: bool = True,
) -> None: ) -> None:
if solver is None: if solver is None:
solver = GurobiPyomoSolver() solver = GurobiPyomoSolver()
if extractor is None: if extractor is None:
extractor = FeaturesExtractor() extractor = FeaturesExtractor(
with_sa=extract_sa,
with_lhs=extract_lhs,
)
assert isinstance(solver, InternalSolver) assert isinstance(solver, InternalSolver)
self.components: Dict[str, Component] = {} self.components: Dict[str, Component] = {}
self.internal_solver: Optional[InternalSolver] = None self.internal_solver: Optional[InternalSolver] = None