test.py
520 Bytes
import config
from dataset import load_dataset
from tensorflow.keras.models import load_model
import tensorflow as tf
pairData, pairLabels = load_dataset(config.DATASET_PATH)
print("Loaded Dataset")
X1 = pairData[:, 0].tolist()
X2 = pairData[:, 1].tolist()
Label = pairLabels[:].tolist()
X1 = tf.convert_to_tensor(X1)
X2 = tf.convert_to_tensor(X2)
Label = tf.convert_to_tensor(Label)
model = load_model(config.MODEL_PATH)
result = model.evaluate([X1, X2], Label, batch_size=64)
print("test loss, test acc:", result)