Spaces:
Sleeping
Sleeping
fix errors
Browse files
app.py
CHANGED
@@ -208,10 +208,10 @@ def load_data():
|
|
208 |
|
209 |
predictions, labels = get_predictions_labels(model, dataloader)
|
210 |
|
211 |
-
|
212 |
-
|
213 |
|
214 |
-
|
215 |
reference_labels = [x.split('Feedback:', 1)[0].strip() for x in labels]
|
216 |
|
217 |
rouge_score = rouge.compute(predictions=predicted_feedback, references=reference_feedback)['rouge2']
|
@@ -228,7 +228,7 @@ def load_data():
|
|
228 |
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)
|
229 |
|
230 |
df = pd.concat([df, new_row])
|
231 |
-
|
232 |
|
233 |
dataframe = load_data()
|
234 |
|
|
|
208 |
|
209 |
predictions, labels = get_predictions_labels(model, dataloader)
|
210 |
|
211 |
+
predicted_feedback = extract_feedback(predictions)
|
212 |
+
predicted_labels = extract_labels(predictions)
|
213 |
|
214 |
+
reference_feedback = [x.split('Feedback:', 1)[1].strip() for x in labels]
|
215 |
reference_labels = [x.split('Feedback:', 1)[0].strip() for x in labels]
|
216 |
|
217 |
rouge_score = rouge.compute(predictions=predicted_feedback, references=reference_feedback)['rouge2']
|
|
|
228 |
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)
|
229 |
|
230 |
df = pd.concat([df, new_row])
|
231 |
+
return df
|
232 |
|
233 |
dataframe = load_data()
|
234 |
|