Update evaluation.py

evaluation.py

@@ -89,7 +89,12 @@ def adherence(response, relevant_documents):
 
 # Step 6: Compute RMSE for metrics
 def compute_rmse(predicted_values, ground_truth_values):
-    return np.sqrt(mean_squared_error(ground_truth_values, predicted_values))
+    # Ensure that both predicted_values and ground_truth_values are numeric
+    if all(isinstance(i, (int, float)) for i in predicted_values) and all(isinstance(i, (int, float)) for i in ground_truth_values):
+        return np.sqrt(mean_squared_error(ground_truth_values, predicted_values))
+    else:
+        print("Invalid input for RMSE calculation. Ensure all values are numeric.")
+        return None
     
 def retrieve_ground_truths(question, dataset):
     """Retrieve the ground truth answer for a given question from the dataset."""