diff options
| author | nunzip <np.scarh@gmail.com> | 2019-02-27 19:20:17 +0000 |
|---|---|---|
| committer | nunzip <np.scarh@gmail.com> | 2019-02-27 19:20:17 +0000 |
| commit | f5e7e167119dff9e2bae122f44fc2172b0cae14b (patch) | |
| tree | e13e85aa1c49de19d4f2c77db9fadd8c3a834a23 /lenet.py | |
| parent | ff33b976f5d1846041b472c87e0f5d5cf6e79529 (diff) | |
| download | e4-gan-f5e7e167119dff9e2bae122f44fc2172b0cae14b.tar.gz e4-gan-f5e7e167119dff9e2bae122f44fc2172b0cae14b.tar.bz2 e4-gan-f5e7e167119dff9e2bae122f44fc2172b0cae14b.zip | |
Fix y_test format bug
Diffstat (limited to 'lenet.py')
| -rw-r--r-- | lenet.py | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -20,6 +20,7 @@ def import_mnist(): X_test, y_test = mnist.test.images, mnist.test.labels y_train = keras.utils.to_categorical(y_train, 10) y_validation = keras.utils.to_categorical(y_validation, 10) + y_test = keras.utils.to_categorical(y_test, 10) return X_train, y_train, X_validation, y_validation, X_test, y_test def plot_images(images, cls_true, cls_pred=None): |