# Train the model
model.fit(x_train, y_train, epochs=5, callbacks=[tensorboard_callback])

        __
        __