import numpy as np

y_pred = []  
y_label = []  

for images, labels in test_data:
    batch_pred = tf.argmax(VGG_model(images, training=False), axis=1).numpy()  # Get predictions
    y_pred.extend(batch_pred)  

    batch_labels = tf.argmax(labels, axis=1).numpy()
    y_label.extend(batch_labels)  

y_pred = np.array(y_pred)
y_label = np.array(y_label)

        __
        __