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

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

8
miplearn/solvers/learning.py
View File

@@ -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: