.

2017-01-05 04:56:35 -08:00 · 2017-01-05 04:56:35 -08:00 · f7e3cfceb1
parent 15d053789e
commit f7e3cfceb1
1 changed files with 1 additions and 1 deletions
--- a/resnet.py
+++ b/resnet.py
@ -52,7 +52,7 @@ use_image_generator = True
 def prepare(X, y):
    X = X.reshape(X.shape[0], 1, width, height).astype('float32') / 255
    # convert class vectors to binary class matrices
-    Y = to_categorical(y_train, nb_classes)
+    Y = to_categorical(y, nb_classes)
    return X, Y

 # the data, shuffled and split between train and test sets