Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,7 +113,7 @@ with st.sidebar:
|
|
| 113 |
)
|
| 114 |
callbacks.append(es)
|
| 115 |
|
| 116 |
-
hist=model.fit(x_train,y_train,epochs=
|
| 117 |
|
| 118 |
# --- Neural Network Diagram ---
|
| 119 |
st.subheader("Neural Network Architecture")
|
|
|
|
| 113 |
)
|
| 114 |
callbacks.append(es)
|
| 115 |
|
| 116 |
+
hist=model.fit(x_train,y_train,epochs=epochs,batch_size=train_size,validation_data=(x_test, y_test),verbose=False)
|
| 117 |
|
| 118 |
# --- Neural Network Diagram ---
|
| 119 |
st.subheader("Neural Network Architecture")
|