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

LearningSolver: Silence logging during parallel_solve

Browse Source
This commit is contained in:
Alinson S. Xavier
2020-10-05 11:27:43 -05:00
parent 92615958fb
commit 0cf88ded89
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
miplearn/solvers/learning.py
View File

@@ -238,13 +238,13 @@ class LearningSolver:
label="Solve"):
self.internal_solver = None
self._silence_miplearn_logger()
SOLVER[0] = self
INSTANCES[0] = instances
p_map_results = p_map(_parallel_solve,
list(range(len(instances))),
num_cpus=n_jobs,
desc=label)
results = [p["Results"] for p in p_map_results]
for (idx, r) in enumerate(p_map_results):
instances[idx].solution = r["Solution"]
@@ -255,7 +255,7 @@ class LearningSolver:
instances[idx].found_violated_lazy_constraints = r["Violated lazy constraints"]
#instances[idx].found_violated_user_cuts = r["Violated user cuts"]
instances[idx].solver_log = r["Results"]["Log"]
self._restore_miplearn_logger()
return results
def fit(self, training_instances):
@@ -268,6 +268,15 @@ class LearningSolver:
name = component.__class__.__name__
self.components[name] = component
def _silence_miplearn_logger(self):
miplearn_logger = logging.getLogger("miplearn")
self.prev_log_level = miplearn_logger.getEffectiveLevel()
miplearn_logger.setLevel(logging.WARNING)
def _restore_miplearn_logger(self):
miplearn_logger = logging.getLogger("miplearn")
miplearn_logger.setLevel(self.prev_log_level)
def __getstate__(self):
self.internal_solver = None
return self.__dict__