run_log_dir = experiment_log_dir + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=run_log_dir, histogram_freq=1)
model.fit(x_train, y_train, epochs=5, batch_size=64, callbacks=[tensorboard_callback])
TensorFlow tutorial - MNIST For ML Beginners
This notebook demonstrates how to use TensorFlow on the Spark driver node to fit a neural network on MNIST handwritten digit recognition data.
Prerequisites:
The content of this notebook is adapted from TensorFlow project under Apache 2.0 license with slight modification to run on Databricks. Thanks to the developers of TensorFlow for this example!