Add experiment title and model version to reports

This commit is contained in:
2021-12-17 23:05:14 +01:00
parent 4a3256c5b7
commit 9300043cfa
5 changed files with 49 additions and 7 deletions

View File

@@ -49,6 +49,9 @@ def parse_arguments():
ap.add_argument(
"-f", "--paramfile", type=bool, required=False, default=False
)
ap.add_argument(
"--title", type=str, required=True, help="experiment title"
)
ap.add_argument(
"-q",
"--quiet",
@@ -85,6 +88,7 @@ def parse_arguments():
args.hyperparameters,
args.paramfile,
args.report,
args.title,
)
@@ -98,6 +102,7 @@ def parse_arguments():
hyperparameters,
paramfile,
report,
experiment_title,
) = parse_arguments()
job = Experiment(
score_name=score,
@@ -108,6 +113,7 @@ job = Experiment(
hyperparams_file=paramfile,
progress_bar=not quiet,
platform=platform,
title=experiment_title,
folds=folds,
)
job.do_experiment()