Skip extracting features if already computed

master
Alinson S. Xavier 5 years ago
parent 5aa434b439
commit 74ceb776c3

@ -155,8 +155,9 @@ class LearningSolver:
# Extract features # Extract features
# ------------------------------------------------------- # -------------------------------------------------------
logger.info("Extracting features...") if instance.features is None:
FeaturesExtractor(self.internal_solver).extract(instance) logger.info("Extracting features...")
FeaturesExtractor(self.internal_solver).extract(instance)
callback_args = ( callback_args = (
self, self,
@ -381,10 +382,11 @@ class LearningSolver:
return stats return stats
def fit(self, training_instances: List[Instance]) -> None: def fit(self, training_instances: List[Instance]) -> None:
logger.debug("Fitting...")
if len(training_instances) == 0: if len(training_instances) == 0:
logger.warn("Empty list of training instances provided. Skipping.")
return return
for component in self.components.values(): for component in self.components.values():
logger.info(f"Fitting {component.__class__.__name__}...")
component.fit(training_instances) component.fit(training_instances)
def _add_component(self, component: Component) -> None: def _add_component(self, component: Component) -> None:

Loading…
Cancel
Save