Metadata-Version: 2.0
Name: gpkg.mnist
Version: 0.3.0.dev5
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
Keywords: mnist cnn softmax images model
Platform: UNKNOWN



Models
######

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

*CNN classifier for MNIST*

Operations
==========

train
^^^^^

*Train the CNN*

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-softmax
#############

*Softmax regression classifier for MNIST*

Operations
==========

train
^^^^^

*Train the softmax regression*

Flags
-----

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

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



