add decimal to normalization value

carpentries-incubator · May 11, 2024 · e4d68a3 · e4d68a3
1 parent ad2427e
commit e4d68a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))