View source on GitHub |
APIs to train an on-device recommendation model.
Demo code:
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.