MCK-02 commited on
Commit
83efb6a
1 Parent(s): 664aff5

fix syntax

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- new_row = pd.Dataframe("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
-
 
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