Add score hyperparameter

This commit is contained in:
2021-09-27 16:32:17 +02:00
parent 7a4409bb1f
commit 50cbc95919
7 changed files with 137 additions and 65 deletions

View File

@@ -8,6 +8,13 @@ from Results import Report
def parse_arguments():
ap = argparse.ArgumentParser()
ap.add_argument(
"-s",
"--score",
type=str,
required=True,
help="score name {accuracy, f1_macro, ...}",
)
ap.add_argument(
"-P",
"--platform",
@@ -55,6 +62,7 @@ def parse_arguments():
)
args = ap.parse_args()
return (
args.score,
args.model,
args.n_folds,
args.platform,
@@ -66,6 +74,7 @@ def parse_arguments():
(
score,
model,
folds,
platform,
@@ -75,6 +84,7 @@ def parse_arguments():
report,
) = parse_arguments()
job = Experiment(
score_name=score,
model_name=model,
datasets=Datasets(),
hyperparams_dict=hyperparameters,