| ## Chargement et normalisation des données | ## Chargement et normalisation des données | ||||
| resnet18 = tf.keras.datasets.cifar10 | resnet18 = tf.keras.datasets.cifar10 | ||||
| (train_images, train_labels), (test_images, test_labels) = resnet18.load_data() | (train_images, train_labels), (test_images, test_labels) = resnet18.load_data() | ||||
| train_images = train_images.astype('float32') | |||||
| test_images = test_images.astype('float32') | |||||
| from sklearn.model_selection import train_test_split | from sklearn.model_selection import train_test_split | ||||
| train_images, val_images, train_labels, val_labels = train_test_split(train_images,train_labels, test_size = 0.2,shuffle = True) | train_images, val_images, train_labels, val_labels = train_test_split(train_images,train_labels, test_size = 0.2,shuffle = True) | ||||
| history = model.fit(train_images, | history = model.fit(train_images, | ||||
| train_labels, | train_labels, | ||||
| batch_size=64, | |||||
| epochs=4, | |||||
| batch_size=256, | |||||
| epochs=50, | |||||
| validation_data=(val_images, val_labels), | validation_data=(val_images, val_labels), | ||||
| ) | ) | ||||