Fix issue with EMNIST

machinecurve · Jan 9, 2020 · 1ef521f · 1ef521f
1 parent bab25b1
commit 1ef521f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extra_keras_datasets/emnist.py b/extra_keras_datasets/emnist.py
@@ -52,4 +52,4 @@ def load_data(path='emnist_matlab.npz', type='balanced'):
     input_test = input_test.reshape((input_test.shape[0], 28, 28), order='F')
 
     # Return data
-    return (input_train, target_train), (input_test, target_train)
+    return (input_train, target_train), (input_test, target_test)