Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

Module: tflite_model_maker.recommendation

APIs to train an on-device recommendation model.

Demo code:

https://github.com/tensorflow/examples/blob/master/tensorflow_examples/lite/model_maker/demo/recommendation_demo.py

Modules

spec module: APIs for recommendation specifications.

Classes

class DataLoader: Recommendation data loader.

class ModelSpec: Recommendation model spec.

class Recommendation: Recommendation task class.

Functions

create(...): Loads data and train the model for recommendation.