confusion_matrix = tf.math.confusion_matrix(
    y_label,
    y_pred,
    num_classes=len(class_names)
)
print("Confusion Matrix:")
print(confusion_matrix.numpy())

        __
        __