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['Context Relevance']
37
- predicted_utilization = metrics['Context Utilization']
38
- predicted_adherence = metrics['Adherence']
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.5 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
 
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