Ilia Tambovtsev commited on
Commit
a2dc318
·
1 Parent(s): afb9c30

feat: log n questions

Browse files
Files changed (1) hide show
  1. src/eval/eval_mlflow.py +1 -0
src/eval/eval_mlflow.py CHANGED
@@ -663,5 +663,6 @@ class RAGEvaluatorMlflow:
663
  if values:
664
  mean_value = sum(values) / len(values)
665
  mlflow.log_metric(f"mean_{name}", mean_value)
 
666
  mlflow.log_metric(f"error_rate", n_errors / len(questions_df))
667
  self._logger.info(f"Mean {name}: {mean_value:.3f}")
 
663
  if values:
664
  mean_value = sum(values) / len(values)
665
  mlflow.log_metric(f"mean_{name}", mean_value)
666
+ mlflow.log_metric(f"n_questions", len(questions_df))
667
  mlflow.log_metric(f"error_rate", n_errors / len(questions_df))
668
  self._logger.info(f"Mean {name}: {mean_value:.3f}")