diff --git a/read_cifar.py b/read_cifar.py index 362ba3d688ac0ffefe9dc0d10dc90cde0f171a1b..7b24a902e2ef69c1e7c485517612e3c3606eac61 100644 --- a/read_cifar.py +++ b/read_cifar.py @@ -3,7 +3,6 @@ import numpy as np from six.moves import cPickle as pickle import os import platform -from sklearn.utils import shuffle #Defining the classes contained in the CIFAR-10 dataset classes = ('plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck')