diff --git a/benchmark/scripts/be_best b/benchmark/scripts/be_best index 5504770..b6a8312 100755 --- a/benchmark/scripts/be_best +++ b/benchmark/scripts/be_best @@ -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] diff --git a/benchmark/scripts/be_main b/benchmark/scripts/be_main index 60fa2b7..cb9cfea 100755 --- a/benchmark/scripts/be_main +++ b/benchmark/scripts/be_main @@ -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",