fonciton 1 finie

read_cifar.py

-import numpy
+import numpy as np
 import pickle
-def unpickle(file):
+def read_cifar_batch(file):
     with open(file, 'rb') as fo:
         dict = pickle.load(fo, encoding='bytes')
-    return dict
+        data = np.array(dict[b'data']).astype('float32')
+        labels = np.array(dict[b'labels']).astype('int64')
+    return data, labels
 
-dict1= unpickle("data/cifar-10-batches-py/data_batch_1")
+vect1= read_cifar_batch("data/cifar-10-batches-py/data_batch_1")
 
-data = [dict1[key] for key in dict1]
+#print(vect1)
 
-print(len(data[1]))
-
-#def read_cifar_batch(path) :
+#def read_cifar
\ No newline at end of file