Skip to content

Commit 3bb060f

Browse files
Update demo01_imageClassificationFromScratch.py
1 parent 7ee9664 commit 3bb060f

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

Article06 - Keras/demo01_imageClassificationFromScratch.py

-3
Original file line numberDiff line numberDiff line change
@@ -132,11 +132,9 @@ def make_model(input_shape, num_classes):
132132
outputs = layers.Dense(units, activation=activation)(x)
133133
return keras.Model(inputs, outputs)
134134

135-
136135
model = make_model(input_shape=image_size + (3,), num_classes=2)
137136
keras.utils.plot_model(model, show_shapes=True)
138137

139-
140138
# Train the model
141139
epochs = 50
142140

@@ -151,4 +149,3 @@ def make_model(input_shape, num_classes):
151149
model.fit(
152150
train_ds, epochs=epochs, callbacks=callbacks, validation_data=val_ds,
153151
)
154-

0 commit comments

Comments
 (0)