# Compute confusion matrix
confusion_matrix = tf.math.confusion_matrix(
    y_true,
    y_pred,
    num_classes=config["num_classes"]
)

# Display confusion matrix
print("Confusion Matrix:")
print(confusion_matrix.numpy())

        __
        __