From 6969f2ffd2c3658b050eb1d8fb53b1684b0d00be Mon Sep 17 00:00:00 2001 From: "Alinson S. Xavier" Date: Tue, 29 Jun 2021 07:52:04 -0500 Subject: [PATCH] Measure time extracting features --- miplearn/solvers/learning.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/miplearn/solvers/learning.py b/miplearn/solvers/learning.py index 7864995..a6eb395 100644 --- a/miplearn/solvers/learning.py +++ b/miplearn/solvers/learning.py @@ -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