Mlflow можно записывать путаницу в матрицу каждый шаг?

Question

Можно ли с помощью mlflow регистрировать путаницу с каждым шагом, как с простой метрикой? Если это возможно, у него есть такая визуализация?

Answer 1

Для каждого возможного прогона вы можете получить индивидуальные значения путаницы c значения

# get confusion matrix values
conf_matrix = confusion_matrix(y_test,y_pred)
true_positive = conf_matrix[0][0]
true_negative = conf_matrix[1][1]
false_positive = conf_matrix[0][1]
false_negative = conf_matrix[1][0]

mlflow.log_metric("true_positive", true_positive)
mlflow.log_metric("true_negative", true_negative)
mlflow.log_metric("false_positive", false_positive)
mlflow.log_metric("false_negative", false_negative)

And then log_artifact(<your_plot>, "confusion_matrix")