Spaces:
Build error
Build error
Ilia Tambovtsev commited on
Commit ·
a2dc318
1
Parent(s): afb9c30
feat: log n questions
Browse files- 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}")
|