CosmickVisions commited on
Commit
bcf8c81
·
verified ·
1 Parent(s): 2de8c50

Update app.py

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