Question

A data scientist has trained a scikit-learn classifier inside an MLflow run and generated a confusion matrix plot saved locally as 'confusion_matrix.png'. They want this image to be stored with the run so teammates can view it in the MLflow UI under the run's artifacts. Which MLflow API call should they use inside the active run?

Accepted Answer

mlflow.log_artifact('confusion_matrix.png')

Answer

mlflow.log_metric('confusion_matrix', 'confusion_matrix.png')

Answer

mlflow.log_param('confusion_matrix', 'confusion_matrix.png')

Answer

mlflow.set_tag('confusion_matrix', 'confusion_matrix.png')