We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 5e30903 + bb0007e commit 0cb1ba2Copy full SHA for 0cb1ba2
fastchat/serve/monitor/classify/display_score.py
@@ -1,6 +1,7 @@
1
import pandas as pd
2
import argparse
3
import os
4
+from pathlib import Path
5
from glob import glob
6
from sklearn.metrics import recall_score, precision_score
7
@@ -40,8 +41,7 @@
40
41
recall = recall_score(y_pred=test.pred, y_true=test.label)
42
precision = precision_score(y_pred=test.pred, y_true=test.label)
43
- # print(f"Model: {output.model[0]}")
44
- print(f"Model: {os.path.splitext(os.path.basename(file))[0]}")
+ print(f"Classifier: {Path(file).stem}")
45
print(f"Accuracy: {round(accuracy, 3)}")
46
print(f"Precision: {round(precision, 3)}")
47
print(f"Recall: {round(recall, 3)}")
0 commit comments