Parallel processing

This commit is contained in:
2021-04-13 19:28:18 -05:00
parent bec7dae6d9
commit 77b10b9609
9 changed files with 45 additions and 25 deletions

View File

@@ -391,11 +391,19 @@ class LearningSolver:
self._restore_miplearn_logger()
return stats
def fit(self, training_instances: List[Instance]) -> None:
def fit(
self,
training_instances: List[Instance],
n_jobs: int = 1,
) -> None:
if len(training_instances) == 0:
logger.warning("Empty list of training instances provided. Skipping.")
return
Component.fit_multiple(self.components, training_instances)
Component.fit_multiple(
self.components,
training_instances,
n_jobs=n_jobs,
)
def _add_component(self, component: Component) -> None:
name = component.__class__.__name__