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

remove unnecessary assert

Browse source
This commit is contained in:
Connor Olding 2018-03-14 17:20:44 +01:00
parent 3a187c920c
commit 793b6139f2
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
mnists/__init__.py
View file

@ -218,10 +218,8 @@ def prepare(dataset="mnist", return_floats=True, return_onehot=True,
test_images_data = test_images_data.transpose(0,1,3,2)
if return_floats: # TODO: better name.
train_images_data = train_images_data.astype(np.float32) / np.float32(255)
test_images_data = test_images_data.astype(np.float32) / np.float32(255)
assert train_images_data.dtype == 'float32'
assert test_images_data.dtype == 'float32'
train_images_data = train_images_data.astype(np.float32) / 255
test_images_data = test_images_data.astype(np.float32) / 255
# emnist_letters uses labels indexed from 1 instead of the usual 0.
# the onehot function assumes labels are contiguous from 0.