Vertex AI Experiments

classification_metrics.py

from sklearn.metrics import confusion_matrix

labels = ["ant", "bird", "cat"]
y_true = ["cat", "ant", "cat", "cat", "ant", "bird"]
y_pred = ["ant", "ant", "cat", "cat", "ant", "cat"]

classification_metrics = {
        "matrix": confusion_matrix(y_true, y_pred, labels=labels).tolist(),
        "labels": labels
        }

print(classification_metrics)

aiplatform.log_classification_metrics(
        labels=classification_metrics["labels"],
        matrix=classification_metrics["matrix"],
        display_name="my-confusion-matrix",
    )