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

Convert LPSolveStats into dataclass

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-11 08:41:50 -05:00
parent 6afdf2ed55
commit 945f6a091c
7 changed files with 24 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
miplearn/solvers/learning.py
View File

@@ -178,10 +178,10 @@ class LearningSolver:
logger.info("Solving root LP relaxation...")
lp_stats = self.internal_solver.solve_lp(tee=tee)
stats.update(cast(LearningSolveStats, lp_stats))
stats.update(cast(LearningSolveStats, lp_stats.__dict__))
training_sample.lp_solution = self.internal_solver.get_solution()
training_sample.lp_value = lp_stats["LP value"]
training_sample.lp_log = lp_stats["LP log"]
training_sample.lp_value = lp_stats.lp_value
training_sample.lp_log = lp_stats.lp_log
logger.debug("Running after_solve_lp callbacks...")
for component in self.components.values():
@@ -240,8 +240,6 @@ class LearningSolver:
lazy_cb=lazy_cb,
)
stats.update(cast(LearningSolveStats, mip_stats))
if training_sample.lp_value is not None:
stats["LP value"] = training_sample.lp_value
stats["Solver"] = "default"
stats["Gap"] = self._compute_gap(
ub=stats["Upper bound"],