tflite_model_maker.recommendation.spec.ModelHParams

Class to hold parameters for model architecture configuration.

hidden_layer_dims List of hidden layer dimensions.
eval_top_k Top k to evaluate.
conv_num_filter_ratios Number of filter ratios for the Conv1D layer.
conv_kernel_size Size of the Conv1D layer kernel size.
lstm_num_units Number of units for the LSTM layer.
num_predictions Number of predictions to return with serving mode, which has default value 10.

Methods

__eq__

Method generated by attrs for class ModelConfig.

__ge__

Method generated by attrs for class ModelConfig.

__gt__

Method generated by attrs for class ModelConfig.

__le__

Method generated by attrs for class ModelConfig.

__lt__

Method generated by attrs for class ModelConfig.

__ne__

Method generated by attrs for class ModelConfig.