Get started with TensorFlow Keras in Databricks

Note

The managed MLflow integration with Databricks on Google Cloud requires Databricks Runtime for Machine Learning 8.1 or above.

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.

Requirements

Databricks Runtime ML

Example notebook

Get started with TensorFlow Keras notebook

Open notebook in new tab