Spaces:
Runtime error
Runtime error
Commit
·
d3c9972
1
Parent(s):
4b34461
Update app.py
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ model_9.compile(loss="categorical_crossentropy", # changed to categorical_crosse
|
|
126 |
|
127 |
# Fit the model
|
128 |
history_9 = model_9.fit(train_data, # now 10 different classes
|
129 |
-
epochs=
|
130 |
steps_per_epoch=len(train_data),
|
131 |
validation_data=test_data,
|
132 |
validation_steps=len(test_data))
|
|
|
126 |
|
127 |
# Fit the model
|
128 |
history_9 = model_9.fit(train_data, # now 10 different classes
|
129 |
+
epochs=16,
|
130 |
steps_per_epoch=len(train_data),
|
131 |
validation_data=test_data,
|
132 |
validation_steps=len(test_data))
|