Spaces:
Sleeping
Sleeping
fix syntax
Browse files
app.py
CHANGED
|
@@ -224,8 +224,9 @@ def load_data():
|
|
| 224 |
f1_weighted_value = f1_score(reference_labels_np, predicted_labels, average='weighted')
|
| 225 |
f1_macro_value = f1_score(reference_labels_np, predicted_labels, average='macro', labels=['Incorrect', 'Partially correct', 'Correct'])
|
| 226 |
|
| 227 |
-
|
| 228 |
-
|
|
|
|
| 229 |
df = pd.concat([df, new_row])
|
| 230 |
return df
|
| 231 |
|
|
|
|
| 224 |
f1_weighted_value = f1_score(reference_labels_np, predicted_labels, average='weighted')
|
| 225 |
f1_macro_value = f1_score(reference_labels_np, predicted_labels, average='macro', labels=['Incorrect', 'Partially correct', 'Correct'])
|
| 226 |
|
| 227 |
+
new_row_data = {"Model": get_model(ds), "Dataset": ds, "SacreBLEU": bleu_score, "ROUGE-2": rouge_score, "METEOR": meteor_score, "BERTScore": bert_score, "Accuracy": accuracy_value, "Weighted F1": f1_weighted_value, "Macro F1": f1_macro_value}
|
| 228 |
+
new_row = pd.DataFrame(new_row_data)
|
| 229 |
+
|
| 230 |
df = pd.concat([df, new_row])
|
| 231 |
return df
|
| 232 |
|