MnistClassifierHparams#
- class composer.models.MnistClassifierHparams(initializers=<factory>, num_classes=None)[source]#
YAHP interface for
mnist_model()
.- Parameters
num_classes (int, optional) โ The number of classes. Needed for classification tasks. Default: 10.
initializers (List[Initializer], optional) โ Initializers for the model.
None
for no initialization. Default:None
.