From 74ceb776c3cc4688f3c3fadf8ca691fd6632644c Mon Sep 17 00:00:00 2001 From: Alinson S Xavier Date: Fri, 9 Apr 2021 07:51:33 -0500 Subject: [PATCH] Skip extracting features if already computed --- miplearn/solvers/learning.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/miplearn/solvers/learning.py b/miplearn/solvers/learning.py index 351d863..cce10f0 100644 --- a/miplearn/solvers/learning.py +++ b/miplearn/solvers/learning.py @@ -155,8 +155,9 @@ class LearningSolver: # Extract features # ------------------------------------------------------- - logger.info("Extracting features...") - FeaturesExtractor(self.internal_solver).extract(instance) + if instance.features is None: + logger.info("Extracting features...") + FeaturesExtractor(self.internal_solver).extract(instance) callback_args = ( self, @@ -381,10 +382,11 @@ class LearningSolver: return stats def fit(self, training_instances: List[Instance]) -> None: - logger.debug("Fitting...") if len(training_instances) == 0: + logger.warn("Empty list of training instances provided. Skipping.") return for component in self.components.values(): + logger.info(f"Fitting {component.__class__.__name__}...") component.fit(training_instances) def _add_component(self, component: Component) -> None: