Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -735,7 +735,7 @@ elif app_mode == "Model Training":
|
|
735 |
model_options = ["Logistic Regression", "Decision Tree", "Random Forest", "Gradient Boosting", "SVM", "Neural Network", "KNN", "Naive Bayes"]
|
736 |
model_name = st.selectbox("Select Model", model_options, help="Choose a model.")
|
737 |
|
738 |
-
|
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)
|
|
|
735 |
model_options = ["Logistic Regression", "Decision Tree", "Random Forest", "Gradient Boosting", "SVM", "Neural Network", "KNN", "Naive Bayes"]
|
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)
|