Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -736,14 +736,14 @@ elif app_mode == "Model Training":
|
|
736 |
model_name = st.selectbox("Select Model", model_options, help="Choose a model.")
|
737 |
|
738 |
if model_name == "Gradient Boosting":
|
739 |
-
|
740 |
-
|
741 |
-
|
742 |
-
|
743 |
-
|
744 |
-
|
745 |
-
|
746 |
-
|
747 |
elif model_name == "Neural Network":
|
748 |
hidden_layers = st.slider("Number of Hidden Layers", 1, 5, 2)
|
749 |
neurons_per_layer = st.slider("Neurons per Layer", 10, 200, 50)
|
|
|
736 |
model_name = st.selectbox("Select Model", model_options, help="Choose a model.")
|
737 |
|
738 |
if model_name == "Gradient Boosting":
|
739 |
+
learning_rate = st.slider("Learning Rate", 0.01, 1.0, 0.1)
|
740 |
+
n_estimators = st.slider("Number of Estimators", 10, 200, 100)
|
741 |
+
max_depth = st.slider("Max Depth", 3, 20, 10)
|
742 |
+
hyperparams = {
|
743 |
+
'learning_rate': learning_rate,
|
744 |
+
'n_estimators': n_estimators,
|
745 |
+
'max_depth': max_depth
|
746 |
+
}
|
747 |
elif model_name == "Neural Network":
|
748 |
hidden_layers = st.slider("Number of Hidden Layers", 1, 5, 2)
|
749 |
neurons_per_layer = st.slider("Neurons per Layer", 10, 200, 50)
|