From cbc9c4d7e4a62918061b54e0e77b7d6c88e4a232 Mon Sep 17 00:00:00 2001 From: Connor Olding Date: Sat, 24 Mar 2018 08:14:29 +0100 Subject: [PATCH] fix typo: squeeze_layers -> squeeze_labels --- mnists/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mnists/__init__.py b/mnists/__init__.py index 6326b93..67ea989 100644 --- a/mnists/__init__.py +++ b/mnists/__init__.py @@ -148,7 +148,7 @@ def batch_flatten(x): def prepare(dataset="mnist", return_floats=True, return_onehot=True, - flatten=False, squeeze_layers=True, check_integrity=True): + flatten=False, squeeze_labels=True, check_integrity=True): if dataset not in metadata.keys(): raise Exception(f"Unknown dataset: {dataset}") @@ -191,7 +191,7 @@ def prepare(dataset="mnist", return_floats=True, return_onehot=True, # emnist_letters uses labels indexed from 1 instead of the usual 0. # the onehot function assumes labels are contiguous from 0. - if squeeze_layers or return_onehot: + if squeeze_labels or return_onehot: train_labels_data = squeeze(train_labels_data) test_labels_data = squeeze(test_labels_data)