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

improve printing

This commit is contained in:
Lorenzo Volpi 2023-11-26 16:32:23 +01:00
parent 5e32582782
commit 56a992f2e9
1 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
quacc/method/model_selection.py
View File

@ -44,9 +44,9 @@ class GridSearchAE(BaseAccuracyEstimator):
self.__check_error(error) self.__check_error(error)
assert isinstance(protocol, AbstractProtocol), "unknown protocol" assert isinstance(protocol, AbstractProtocol), "unknown protocol"
def _sout(self, msg): def _sout(self, msg, level=0):
if self.verbose: if level > 0 or self.verbose:
print(f"[{self.__class__.__name__}]: {msg}") print(f"[{self.__class__.__name__}@{self.model.__class__.__name__}]: {msg}")
def __normalize_params(self, params): def __normalize_params(self, params):
__remap = {} __remap = {}
@ -113,8 +113,10 @@ class GridSearchAE(BaseAccuracyEstimator):
self._sout( self._sout(
f"optimization finished: best params {self.best_params_} (score={self.best_score_:.5f}) " f"optimization finished: best params {self.best_params_} (score={self.best_score_:.5f}) "
f"[took {tend:.4f}s]" f"[took {tend:.4f}s]",
level=1,
) )
log = SubLogger.logger() log = SubLogger.logger()
log.debug( log.debug(
f"[{self.model.__class__.__name__}] " f"[{self.model.__class__.__name__}] "
@ -160,7 +162,7 @@ class GridSearchAE(BaseAccuracyEstimator):
ttime = time() - tinit ttime = time() - tinit
self._sout( self._sout(
f"hyperparams={params}\t got score {score:.5f} [took {ttime:.4f}s]" f"hyperparams={params}\t got score {score:.5f} [took {ttime:.4f}s]",
) )
# if self.timeout > 0: # if self.timeout > 0:
@ -169,12 +171,18 @@ class GridSearchAE(BaseAccuracyEstimator):
# self._sout(f"timeout ({self.timeout}s) reached for config {params}") # self._sout(f"timeout ({self.timeout}s) reached for config {params}")
# score = None # score = None
except ValueError as e: except ValueError as e:
self._sout(f"the combination of hyperparameters {params} is invalid") self._sout(
raise e f"the combination of hyperparameters {params} is invalid. Exception: {e}",
level=1,
)
score = None
# raise e
except Exception as e: except Exception as e:
self._sout(f"something went wrong for config {params}; skipping:") self._sout(
self._sout(f"\tException: {e}") f"something went wrong for config {params}; skipping:"
f"\tException: {e}",
level=1,
)
score = None score = None
return params, score, model return params, score, model