add score names to help in scripts

This commit is contained in:
2022-05-03 10:35:59 +02:00
parent 8be1a6d12f
commit 0f5ca402e2
2 changed files with 11 additions and 3 deletions

View File

@@ -14,7 +14,8 @@ def parse_arguments():
action=EnvDefault,
envvar="score",
required=True,
help="score name {accuracy, f1_micro, f1_macro, all}",
help="score name {accuracy, f1-micro, f1-macro, f1-weighted, "
"roc-auc-ovr, all}",
)
args = ap.parse_args()
return (args.score,)
@@ -22,7 +23,13 @@ def parse_arguments():
(score,) = parse_arguments()
all_metrics = ["accuracy", "f1-macro", "f1-micro"]
all_metrics = [
"accuracy",
"f1-macro",
"f1-micro",
"f1-weighted",
"roc-auc-ovr",
]
metrics = all_metrics if score == "all" else [score]

View File

@@ -18,7 +18,8 @@ def parse_arguments():
envvar="score",
type=str,
required=True,
help="score name {accuracy, f1_macro, ...}",
help="score name {accuracy, f1-micro, f1-macro, f1-weighted, "
"roc-auc-ovr, all}",
)
ap.add_argument(
"-P",