train.py

from utils import load_data
from model import get_model


print("Loading data...")

(x_train, y_train), (x_test, y_test) = load_data()

print("Data laoded.")
print(x_train.shape)
print(y_train.shape)
print(x_test.shape)
print(y_test.shape)
model = get_model()


batch_size = 128
epochs = 15

model.fit(x_train, y_train, epochs=epochs, batch_size=batch_size)

loss, acc = model.evaluate(x_test, y_test)

print("Test loss: ", loss)
print("Test accuracy: ", acc)

model.save('model.h5')