Attend the Women in ML Symposium on December 7 Register now

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:


spec module: APIs for recommendation specifications.


class DataLoader: Recommendation data loader.

class ModelSpec: Recommendation model spec.

class Recommendation: Recommendation task class.


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