tf.data.Dataset object for MNIST training data.
(directory)
| 112 | |
| 113 | |
| 114 | def train(directory): |
| 115 | """tf.data.Dataset object for MNIST training data.""" |
| 116 | return dataset(directory, 'train-images-idx3-ubyte', |
| 117 | 'train-labels-idx1-ubyte') |
| 118 | |
| 119 | |
| 120 | def test(directory): |