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

Skip extracting features if already computed

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-09 07:51:33 -05:00
parent 5aa434b439
commit 74ceb776c3
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/solvers/learning.py
View File

@@ -155,6 +155,7 @@ class LearningSolver:
# Extract features # Extract features
# ------------------------------------------------------- # -------------------------------------------------------
if instance.features is None:
logger.info("Extracting features...") logger.info("Extracting features...")
FeaturesExtractor(self.internal_solver).extract(instance) FeaturesExtractor(self.internal_solver).extract(instance)
@@ -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: