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

Only include static features in after-load

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-13 16:08:30 -05:00
parent 8f41278713
commit ef7a50e871
11 changed files with 170 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
miplearn/solvers/learning.py
View File

@@ -187,7 +187,10 @@ class LearningSolver:
# Extract features (after-lp)
# -------------------------------------------------------
logger.info("Extracting features (after-lp)...")
features = FeaturesExtractor(self.internal_solver).extract(instance)
features = FeaturesExtractor(self.internal_solver).extract(
instance,
with_static=False,
)
features.extra = {}
features.lp_solve = lp_stats
sample.after_lp = features
@@ -249,7 +252,10 @@ class LearningSolver:
# Extract features (after-mip)
# -------------------------------------------------------
logger.info("Extracting features (after-mip)...")
features = FeaturesExtractor(self.internal_solver).extract(instance)
features = FeaturesExtractor(self.internal_solver).extract(
instance,
with_static=False,
)
features.mip_solve = mip_stats
features.extra = {}
sample.after_mip = features