Skip to content

Commit e60fe2a

Browse files
committed
image minmax scaling added to single model run script
1 parent 6f38dad commit e60fe2a

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

single_model_learn.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,5 +8,7 @@
88

99
(x_train, y_train), (_, _) = datasets.cifar10.load_data()
1010
y_train = utils.to_categorical(y_train, len(np.unique(y_train)))
11+
x_train = x_train.astype(np.float32)
12+
x_train = x_train / 255.0
1113

1214
model.fit(x_train, y_train, batch_size=32, epochs=20, verbose=1)

0 commit comments

Comments
 (0)