Get started with TensorFlow Keras in Databricks

TensorFlow Keras is a deep learning API written in Python that runs on top of the machine learning platform TensorFlow. TensorFlow and Keras are included in Databricks Runtime for Machine Learning. The 10-minute tutorial notebook shows an example of training machine learning models on tabular data with TensorFlow Keras, including using inline TensorBoard.


Databricks Runtime 7.0 ML or above.

Example notebook

Get started with TensorFlow Keras notebook

Open notebook in new tab