# Set the number of training epochs and batch size
epochs = 10
batch_size = 128

# Train the model
model.fit(x_train, y_train, epochs=epochs, batch_size=batch_size, validation_split=0.1)

        __
        __