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

26
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,22 +56,24 @@ class BasicCollector:
model = build_model(data_filename) model = build_model(data_filename)
model.extract_after_load(h5) model.extract_after_load(h5)
# Solve LP relaxation if not self.skip_lp:
relaxed = model.relax() # Solve LP relaxation
relaxed.optimize() relaxed = model.relax()
relaxed.extract_after_lp(h5) relaxed.optimize()
relaxed.extract_after_lp(h5)
# Solve MIP # Solve MIP
model.optimize() model.optimize()
model.extract_after_mip(h5) model.extract_after_mip(h5)
# Add lazy constraints to model if self.write_mps:
if model.lazy_enforce is not None: # Add lazy constraints to model
model.lazy_enforce(model, model.lazy_) if model.lazy_enforce is not None:
model.lazy_enforce(model, model.lazy_)
# Save MPS file # Save MPS file
model.write(mps_filename) model.write(mps_filename)
gzip(mps_filename) gzip(mps_filename)
h5.put_scalar("mip_log", streams[0].getvalue()) h5.put_scalar("mip_log", streams[0].getvalue())