Metadata-Version: 2.0
Name: gpkg.mnist
Version: 0.1.2.dev2
Summary: CNN and softmax regression classifiers for MNIST digits
Home-page: https://github.com/guildai/index/tree/master/mnist
Author: Guild AI
Author-email: packages@guild.ai
License: Apache 2.0
Description-Content-Type: UNKNOWN
Keywords: mnist cnn softmax images model
Platform: UNKNOWN
Requires-Dist: gpkg.mnist-dataset


Models
######

mnist-softmax
#############

*Softmax regression classifier for MNIST*

Operations
==========

train
^^^^^

*Train the softmax regression*

Model flags
===========

**batch-size**
  *Number of images to include in a training batch (default is 100)*

**epochs**
  *Number of epochs to train (default is 10)*

mnist-cnn
#########

*CNN classifier for MNIST*

Operations
==========

train
^^^^^

*Train the CNN*

Model flags
===========

**batch-size**
  *Number of images to include in a training batch (default is 100)*

**epochs**
  *Number of epochs to train (default is 10)*

References
##########

Modelfile: https://github.com/guildai/index/tree/master/mnist/MODELS


