Search⌘ K
AI Features

TF Lite Model Maker Library

Explore how to train custom TensorFlow Lite models using the TF Lite Model Maker library. Learn to prepare data, customize models, evaluate performance, export models, and visualize predictions for mobile deployment.

The TF Lite Model Maker library enables us to train a custom TF Lite model on our datasets. This library supports transfer learning, reducing the required training data in our dataset and the training time. Let’s explore the functionalities offered by the TF Lite Model Maker library to simplify the process of making DL models for mobile and edge devices.

Supported ML tasks

The Model Maker library supports various ML tasks, including:

  • Image classification

  • Text classification

  • Object detection

  • Audio classification

  • Recommendation system ...