From 0f5ca402e299a1781d1f50a84b212b48ecf9a4b9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Tue, 3 May 2022 10:35:59 +0200
Subject: [PATCH] add score names to help in scripts

---
 benchmark/scripts/be_best | 11 +++++++++--
 benchmark/scripts/be_main |  3 ++-
 2 files changed, 11 insertions(+), 3 deletions(-)

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",