From 4d02db9122f757111f94123d6c4df05e43353fef Mon Sep 17 00:00:00 2001
From: BaptisteBrd <75663738+BaptisteBrd@users.noreply.github.com>
Date: Wed, 8 Nov 2023 23:07:22 +0100
Subject: [PATCH] fonciton 1 finie
---
read_cifar.py | 16 ++++++++--------
1 file changed, 8 insertions(+), 8 deletions(-)
diff --git a/read_cifar.py b/read_cifar.py
index 9684ce6..abba3fa 100644
--- a/read_cifar.py
+++ b/read_cifar.py
@@ -1,14 +1,14 @@
-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
--
GitLab