Add GradientBoostingClassifier and fix metrics

2025-08-20 01:45:54 +00:00 · 2022-05-03 15:54:24 +02:00
parent 0f5ca402e2
commit 7f2033193e
12 changed files with 105 additions and 76 deletions
--- a/benchmark/Utils.py
+++ b/benchmark/Utils.py
@@ -3,6 +3,13 @@ import subprocess
 import argparse

 BEST_ACCURACY_STREE = 40.282203
+ALL_METRICS = (
+    "accuracy",
+    "f1-macro",
+    "f1-micro",
+    "f1-weighted",
+    "roc-auc-ovr",
+)


 class Folders: