- name: mnist-softmax
  description: Softmax regression classifier for MNIST
  operations:
    train:
      description: Train the softmax regression
      cmd: softmax --datadir mnist-idx-data --rundir .
      requires: mnist-dataset/idx
  flags:
    batch-size:
      description: Number of images to include in a training batch
      value: 100
    epochs:
      description: Number of epochs to train
      value: 10

- name: mnist-cnn
  description: CNN classifier for MNIST
  operations:
    train:
      description: Train the CNN
      cmd: cnn --datadir mnist-idx-data --rundir .
      requires: mnist-dataset/idx
  flags:
    batch-size:
      description: Number of images to include in a training batch
      value: 100
    epochs:
      description: Number of epochs to train
      value: 10
