From 2cd9adbbcaae2e294ca8a1e542566debde0e178f Mon Sep 17 00:00:00 2001 From: Lorenzo Volpi Date: Wed, 22 Nov 2023 19:21:49 +0100 Subject: [PATCH] all property added to CompEstimator.name --- quacc/evaluation/comp.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/quacc/evaluation/comp.py b/quacc/evaluation/comp.py index e659fb9..344b9fd 100644 --- a/quacc/evaluation/comp.py +++ b/quacc/evaluation/comp.py @@ -29,6 +29,11 @@ class CompEstimatorName_: elif isinstance(e, list): return list(self.ce._CompEstimator__get(e).keys()) + @property + def all(self): + all_keys = list(CompEstimator._CompEstimator__dict.keys()) + return self[all_keys] + class CompEstimatorFunc_: def __init__(self, ce): @@ -78,7 +83,7 @@ CE = CompEstimator() def evaluate_comparison(dataset: Dataset, estimators=None) -> DatasetReport: log = Logger.logger() # with multiprocessing.Pool(1) as pool: - __pool_size = os.cpu_count() // 2 + __pool_size = round(os.cpu_count() * 0.8) with multiprocessing.Pool(__pool_size) as pool: dr = DatasetReport(dataset.name) log.info(f"dataset {dataset.name} [pool size: {__pool_size}]")