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

Measure time extracting features

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-06-29 07:52:04 -05:00
parent 5b4b8adee5
commit 6969f2ffd2
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
miplearn/solvers/learning.py
View File

@@ -4,6 +4,7 @@
import logging
import traceback
import time
from typing import Optional, List, Any, cast, Dict, Tuple
from p_tqdm import p_map
@@ -166,7 +167,12 @@ class LearningSolver:
# Extract features (after-load)
# -------------------------------------------------------
logger.info("Extracting features (after-load)...")
initial_time = time.time()
features = self.extractor.extract(instance, self.internal_solver)
logger.info(
"Features (after-load) extracted in %.2f seconds"
% (time.time() - initial_time)
)
features.extra = {}
sample.after_load = features
@@ -201,11 +207,16 @@ class LearningSolver:
# Extract features (after-lp)
# -------------------------------------------------------
logger.info("Extracting features (after-lp)...")
initial_time = time.time()
features = self.extractor.extract(
instance,
self.internal_solver,
with_static=False,
)
logger.info(
"Features (after-lp) extracted in %.2f seconds"
% (time.time() - initial_time)
)
features.extra = {}
features.lp_solve = lp_stats
sample.after_lp = features
@@ -269,11 +280,16 @@ class LearningSolver:
# Extract features (after-mip)
# -------------------------------------------------------
logger.info("Extracting features (after-mip)...")
initial_time = time.time()
features = self.extractor.extract(
instance,
self.internal_solver,
with_static=False,
)
logger.info(
"Features (after-mip) extracted in %.2f seconds"
% (time.time() - initial_time)
)
features.mip_solve = mip_stats
features.extra = {}
sample.after_mip = features