CosmickVisions commited on
Commit
458b0d9
·
verified ·
1 Parent(s): 39392b6

Update app.py

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