Recommendation model spec.
tflite_model_maker.recommendation.ModelSpec(
input_spec: tflite_model_maker.recommendation.spec.InputSpec
,
model_hparams: tflite_model_maker.recommendation.spec.ModelHParams
)
Args |
input_spec
|
InputSpec, specify data format for input and embedding.
|
model_hparams
|
ModelHParams, specify hparams for model achitecture.
|
Methods
create_model
View source
create_model()
Creates recommendation model based on params.
get_default_quantization_config
View source
get_default_quantization_config()
Gets the default quantization configuration.
Class Variables |
compat_tf_versions
|
[2]
|