Add writing results to disk as they're computed in experiments

This commit is contained in:
2022-03-11 11:30:30 +01:00
parent 33accba9ba
commit d9f1f4b572
2 changed files with 4 additions and 0 deletions

View File

@@ -287,6 +287,7 @@ class Experiment:
position=0, position=0,
disable=not self.progress_bar, disable=not self.progress_bar,
) )
self.duration = 0.0
for name in loop: for name in loop:
loop.set_description(f"{name:30s}") loop.set_description(f"{name:30s}")
X, y = self.datasets.load(name) X, y = self.datasets.load(name)
@@ -296,6 +297,7 @@ class Experiment:
self._init_experiment() self._init_experiment()
self._n_fold_crossval(X, y, hyperparameters) self._n_fold_crossval(X, y, hyperparameters)
self._add_results(name, hyperparameters, samp, feat, n_classes) self._add_results(name, hyperparameters, samp, feat, n_classes)
self._output_results()
self.duration = time.time() - now self.duration = time.time() - now
self._output_results() self._output_results()
if self.progress_bar: if self.progress_bar:

View File

@@ -59,6 +59,8 @@ def parse_arguments():
ap.add_argument( ap.add_argument(
"-m", "-m",
"--model", "--model",
action=EnvDefault,
envvar="model",
type=str, type=str,
required=True, required=True,
help="model name", help="model name",