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

fix downloading and verification of first files

Browse source 
this code would've worked if the first element of `names`
were the prefix i.e. if it acted on `meta` instead of `names`.
pretty embarassing!
This commit is contained in:
Connor Olding 2018-07-02 15:50:05 +02:00
parent c83a306cbe
commit 0aee9edce6
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
mnists/__init__.py
View file

@ -160,7 +160,7 @@ def prepare(dataset="mnist", return_floats=True, return_onehot=True,
prefix, names = meta[0], meta[1:]
names = [os.path.join(prefix, name) for name in names]
train_images, train_labels, test_images, test_labels = names
images_and_labels = names[1:]
images_and_labels = names
logger.debug("Filenames chosen for %s: %s, %s, %s, %s",
dataset, train_images, train_labels, test_images, test_labels)