rahulmishra commited on
Commit
d3c9972
·
1 Parent(s): 4b34461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=17,
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))