Get started with TensorFlow Keras in Databricks

The 10-minute tutorial notebook shows an example of training machine learning models on tabular data with TensorFlow Keras, including using inline TensorBoard.

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.

Requirements

Databricks Runtime ML

Single node and distributed training

To test and migrate single-machine workflows, use a Single Node cluster.

For distributed training options for deep learning, see Distributed training.

Example notebook

Get started with TensorFlow Keras notebook

Open notebook in new tab Copy link for import