vamseelatha2002 commited on
Commit
4d34271
·
verified ·
1 Parent(s): 37e503a

Update evaluation.py

Browse files
Files changed (1) hide show
  1. evaluation.py +1 -1
evaluation.py CHANGED
@@ -189,7 +189,7 @@ def calculate_metrics(question, q_dataset, response, docs, time_taken):
189
  predicted_metrics["rmse"] = "Invalid RMSE calculation"
190
  '''
191
  if isinstance(predicted_metrics_rmse, (int, float)) and isinstance(ground_truth_metrics, (int, float)):
192
- rmse_value = compute_rmse([predicted_metrics_rmse], [ground_truth_metrics])
193
  predicted_metrics["rmse"] = rmse_value # Adding RMSE to metrics
194
  else:
195
  predicted_metrics["rmse"] = "Invalid RMSE calculation"
 
189
  predicted_metrics["rmse"] = "Invalid RMSE calculation"
190
  '''
191
  if isinstance(predicted_metrics_rmse, (int, float)) and isinstance(ground_truth_metrics, (int, float)):
192
+ rmse_value = compute_rmse(predicted_metrics_rmse.values(), ground_truth_metrics.values())
193
  predicted_metrics["rmse"] = rmse_value # Adding RMSE to metrics
194
  else:
195
  predicted_metrics["rmse"] = "Invalid RMSE calculation"