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

Add n_jobs to BenchmarkRunner.fit

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-13 19:30:42 -05:00
parent 77b10b9609
commit e1f32b1798
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
miplearn/benchmark.py
View File

@@ -98,7 +98,7 @@ class BenchmarkRunner:
os.makedirs(os.path.dirname(filename), exist_ok=True) os.makedirs(os.path.dirname(filename), exist_ok=True)
self.results.to_csv(filename) self.results.to_csv(filename)
def fit(self, instances: List[Instance]) -> None: def fit(self, instances: List[Instance], n_jobs: int = 1) -> None:
""" """
Trains all solvers with the provided training instances. Trains all solvers with the provided training instances.
@@ -109,7 +109,7 @@ class BenchmarkRunner:
""" """
for (solver_name, solver) in self.solvers.items(): for (solver_name, solver) in self.solvers.items():
logger.debug(f"Fitting {solver_name}...") logger.debug(f"Fitting {solver_name}...")
solver.fit(instances) solver.fit(instances, n_jobs=n_jobs)
def _silence_miplearn_logger(self) -> None: def _silence_miplearn_logger(self) -> None:
miplearn_logger = logging.getLogger("miplearn") miplearn_logger = logging.getLogger("miplearn")