From 19452c3b91c7f4a1803a0ea12d07bfa9f47fd90f Mon Sep 17 00:00:00 2001 From: Hyeontak Yang Date: Thu, 2 Mar 2017 14:43:11 +0900 Subject: [PATCH] adjust indent and remove not necessary import statements --- dataset/mnist.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/dataset/mnist.py b/dataset/mnist.py index bdea3ea4..a0e84701 100644 --- a/dataset/mnist.py +++ b/dataset/mnist.py @@ -6,7 +6,6 @@ import os.path import gzip import pickle -import os import numpy as np @@ -39,7 +38,7 @@ def _download(file_name): def download_mnist(): for v in key_file.values(): - _download(v) + _download(v) def _load_label(file_name): file_path = dataset_dir + "/" + file_name @@ -118,7 +117,7 @@ def load_mnist(normalize=True, flatten=True, one_hot_label=False): dataset['test_label'] = _change_ont_hot_label(dataset['test_label']) if not flatten: - for key in ('train_img', 'test_img'): + for key in ('train_img', 'test_img'): dataset[key] = dataset[key].reshape(-1, 1, 28, 28) return (dataset['train_img'], dataset['train_label']), (dataset['test_img'], dataset['test_label'])