diff --git a/read_cifar.py b/read_cifar.py index 75954c0da929f019018a35d0eab7260e166c5d4d..f657c0392876393867a7f3158deee4b81a7a272b 100644 --- a/read_cifar.py +++ b/read_cifar.py @@ -29,13 +29,13 @@ def read_cifar(directory_path): for i in range(len(files)): fichier=directory_path+files[i] data_dict=unpickle(fichier) - M=data_dict[b'data'] + D=data_dict[b'data'] L=data_dict[b'labels'] L=np.array(L) - data=np.vstack((X,M)) - labels=np.hstack((Y,L)) - data=X[N:2*N,] - labels=Y[A:,] + data=np.vstack((data,D)) + labels=np.hstack((labels,L)) + data=data[N:2*N,] + labels=labels[A:,] return data,labels def split_dataset(data,labels,split):