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

BasicCollector: Make LP and MPS optional

Browse Source
This commit is contained in:
Alinson S. Xavier
2024-02-01 12:02:23 -06:00
parent 60c7222fbe
commit 5b28595b0b
1 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
miplearn/collectors/basic.py
View File

@@ -16,6 +16,10 @@ from ..parallel import p_umap
class BasicCollector: class BasicCollector:
def __init__(self, skip_lp: bool = False, write_mps: bool = True) -> None:
self.skip_lp = skip_lp
self.write_mps = write_mps
def collect( def collect(
self, self,
filenames: List[str], filenames: List[str],
@@ -52,6 +56,7 @@ class BasicCollector:
model = build_model(data_filename) model = build_model(data_filename)
model.extract_after_load(h5) model.extract_after_load(h5)
if not self.skip_lp:
# Solve LP relaxation # Solve LP relaxation
relaxed = model.relax() relaxed = model.relax()
relaxed.optimize() relaxed.optimize()
@@ -61,6 +66,7 @@ class BasicCollector:
model.optimize() model.optimize()
model.extract_after_mip(h5) model.extract_after_mip(h5)
if self.write_mps:
# Add lazy constraints to model # Add lazy constraints to model
if model.lazy_enforce is not None: if model.lazy_enforce is not None:
model.lazy_enforce(model, model.lazy_) model.lazy_enforce(model, model.lazy_)