File size: 658 Bytes
5c9ce3f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | from tensorflow.keras.models import load_model
from tensorflow.keras.preprocessing import image_dataset_from_directory
import tensorflow as tf
# Load model
model = load_model('model/cat_dog_neither_classifier_new.h5') # Updated filename
# Load test data
test_dataset = image_dataset_from_directory(
'dataset/test_set',
labels='inferred',
label_mode='categorical',
image_size=(224, 224),
batch_size=32
)
# Normalize & prefetch
test_dataset = test_dataset.map(lambda x, y: (x / 255.0, y)).prefetch(tf.data.AUTOTUNE)
# Evaluate
loss, accuracy = model.evaluate(test_dataset)
print(f"Test Accuracy: {accuracy:.4f}")
|