Gourisankar Padihary
commited on
Commit
·
11b4c9f
1
Parent(s):
c8aa52c
Json error fix
Browse files
generator/compute_rmse_auc_roc_metrics.py
CHANGED
@@ -33,9 +33,9 @@ def compute_rmse_auc_roc_metrics(llm, dataset, vector_store, num_question):
|
|
33 |
metrics = generate_metrics(llm, vector_store, query)
|
34 |
|
35 |
# Extract predicted metrics (ensure these are continuous if possible)
|
36 |
-
predicted_relevance = metrics
|
37 |
-
predicted_utilization = metrics
|
38 |
-
predicted_adherence = metrics
|
39 |
|
40 |
# === Handle Continuous Inputs for RMSE ===
|
41 |
relevance_rmse = root_mean_squared_error([ground_truth_relevance], [predicted_relevance])
|
@@ -46,7 +46,7 @@ def compute_rmse_auc_roc_metrics(llm, dataset, vector_store, num_question):
|
|
46 |
binary_ground_truth_relevance = 1 if ground_truth_relevance > 0.5 else 0
|
47 |
#binary_predicted_relevance = 1 if predicted_relevance > 0.5 else 0
|
48 |
|
49 |
-
binary_ground_truth_utilization = 1 if ground_truth_utilization > 0.
|
50 |
#binary_predicted_utilization = 1 if predicted_utilization > 0.5 else 0
|
51 |
|
52 |
#binary_ground_truth_adherence = 1 if ground_truth_adherence > 0.5 else 0
|
|
|
33 |
metrics = generate_metrics(llm, vector_store, query)
|
34 |
|
35 |
# Extract predicted metrics (ensure these are continuous if possible)
|
36 |
+
predicted_relevance = metrics.get('Context Relevance', 0) if metrics else 0
|
37 |
+
predicted_utilization = metrics.get('Context Utilization', 0) if metrics else 0
|
38 |
+
predicted_adherence = metrics.get('Adherence', 0) if metrics else 0
|
39 |
|
40 |
# === Handle Continuous Inputs for RMSE ===
|
41 |
relevance_rmse = root_mean_squared_error([ground_truth_relevance], [predicted_relevance])
|
|
|
46 |
binary_ground_truth_relevance = 1 if ground_truth_relevance > 0.5 else 0
|
47 |
#binary_predicted_relevance = 1 if predicted_relevance > 0.5 else 0
|
48 |
|
49 |
+
binary_ground_truth_utilization = 1 if ground_truth_utilization > 0.2 else 0
|
50 |
#binary_predicted_utilization = 1 if predicted_utilization > 0.5 else 0
|
51 |
|
52 |
#binary_ground_truth_adherence = 1 if ground_truth_adherence > 0.5 else 0
|