MCK-02 commited on
Commit
eb2e963
1 Parent(s): e3e493d

fix errors

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -208,10 +208,10 @@ def load_data():
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,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
- return df
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