Have a question? Connect with the community at the TensorFlow Forum Visit Forum

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

Classes

class DataLoader: Recommendation data loader.

class ModelSpec: Recommendation model spec.

class Recommendation: Recommendation task class.

Functions

BowSpec(...): Creates Recommendation Bag-of-Word (BoW) model spec. See also: tflite_model_maker.recommendation.ModelSpec.

CnnSpec(...): Creates Recommendation CNN model spec. See also: tflite_model_maker.recommendation.ModelSpec.

RnnSpec(...): Creates Recommendation RNN model spec. See also: tflite_model_maker.recommendation.ModelSpec.

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