lorenzo.volpi
/
QuAcc
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

all property added to CompEstimator.name

This commit is contained in:
Lorenzo Volpi 2023-11-22 19:21:49 +01:00
parent 07686b73b6
commit 2cd9adbbca
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
quacc/evaluation/comp.py
View File

@ -29,6 +29,11 @@ class CompEstimatorName_:
elif isinstance(e, list): elif isinstance(e, list):
return list(self.ce._CompEstimator__get(e).keys()) 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_: class CompEstimatorFunc_:
def __init__(self, ce): def __init__(self, ce):
@ -78,7 +83,7 @@ CE = CompEstimator()
def evaluate_comparison(dataset: Dataset, estimators=None) -> DatasetReport: def evaluate_comparison(dataset: Dataset, estimators=None) -> DatasetReport:
log = Logger.logger() log = Logger.logger()
# with multiprocessing.Pool(1) as pool: # 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: with multiprocessing.Pool(__pool_size) as pool:
dr = DatasetReport(dataset.name) dr = DatasetReport(dataset.name)
log.info(f"dataset {dataset.name} [pool size: {__pool_size}]") log.info(f"dataset {dataset.name} [pool size: {__pool_size}]")