From e4d68a35c76a0ca76e57c00bf3ef04c40e867cc4 Mon Sep 17 00:00:00 2001 From: Vlad Dracula Date: Sat, 11 May 2024 12:27:24 +1000 Subject: [PATCH] add decimal to normalization value --- episodes/scripts/02_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/episodes/scripts/02_image.py b/episodes/scripts/02_image.py index 7e558dea..c7994afa 100644 --- a/episodes/scripts/02_image.py +++ b/episodes/scripts/02_image.py @@ -106,7 +106,7 @@ def prepare_dataset(#blank#, #blank#): def prepare_dataset(train_images, train_labels): # normalize the RGB values to be between 0 and 1 - train_images = train_images / 255 + train_images = train_images / 255.0 # one hot encode the training labels train_labels = keras.utils.to_categorical(train_labels, len(class_names))