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

Add mip_stats to after-mip features

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-11 09:14:05 -05:00
parent bd78518c1f
commit 6b15337e4c
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
miplearn/features.py
View File

@@ -13,7 +13,7 @@ import numpy as np
from miplearn.types import Solution, Category from miplearn.types import Solution, Category
if TYPE_CHECKING: if TYPE_CHECKING:
from miplearn.solvers.internal import InternalSolver, LPSolveStats from miplearn.solvers.internal import InternalSolver, LPSolveStats, MIPSolveStats
from miplearn.instance.base import Instance from miplearn.instance.base import Instance
@@ -103,6 +103,7 @@ class Features:
variables: Optional[Dict[str, Variable]] = None variables: Optional[Dict[str, Variable]] = None
constraints: Optional[Dict[str, Constraint]] = None constraints: Optional[Dict[str, Constraint]] = None
lp_solve: Optional["LPSolveStats"] = None lp_solve: Optional["LPSolveStats"] = None
mip_solve: Optional["MIPSolveStats"] = None
class FeaturesExtractor: class FeaturesExtractor:

3
miplearn/solvers/learning.py
View File

@@ -171,6 +171,7 @@ class LearningSolver:
# Solve root LP relaxation # Solve root LP relaxation
# ------------------------------------------------------- # -------------------------------------------------------
lp_stats = None
if self.solve_lp: if self.solve_lp:
logger.debug("Running before_solve_lp callbacks...") logger.debug("Running before_solve_lp callbacks...")
for component in self.components.values(): for component in self.components.values():
@@ -252,6 +253,8 @@ class LearningSolver:
# ------------------------------------------------------- # -------------------------------------------------------
logger.info("Extracting features (after-mip)...") logger.info("Extracting features (after-mip)...")
features = FeaturesExtractor(self.internal_solver).extract(instance) features = FeaturesExtractor(self.internal_solver).extract(instance)
features.lp_solve = lp_stats
features.mip_solve = mip_stats
instance.features_after_mip.append(features) instance.features_after_mip.append(features)
# Add some information to training_sample # Add some information to training_sample