Attend the Women in ML Symposium on December 7 Register now

tflite_model_maker.recommendation.ModelSpec

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

Recommendation model spec.

input_spec InputSpec, specify data format for input and embedding.
model_hparams ModelHParams, specify hparams for model achitecture.

Methods

create_model

View source

Creates recommendation model based on params.

Returns
Keras model.

get_default_quantization_config

View source

Gets the default quantization configuration.

compat_tf_versions [2]