Module: tflite_model_maker.recommendation

Stay organized with collections Save and categorize content based on your preferences.

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.