notwa/gists
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

.

Browse source
This commit is contained in:
Connor Olding 2017-01-05 04:56:35 -08:00
parent 15d053789e
commit f7e3cfceb1
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
resnet.py
View file

@ -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