YEHTUT commited on
Commit
859b26d
·
verified ·
1 Parent(s): 98d77b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -131,7 +131,7 @@ if st.button("Predict"):
131
 
132
  # Make predictions using the pre-trained model
133
  prediction = rf_clf.predict(X_new_scaled)
134
-
135
  # loaded_model = pickle.load(open('Random_Forest.sav', 'rb'))
136
  # prediction = loaded_model.predict(np.array(single_sample))
137
 
 
131
 
132
  # Make predictions using the pre-trained model
133
  prediction = rf_clf.predict(X_new_scaled)
134
+ y_pred_proba = rf_clf.predict_proba(X_new_scaled)
135
  # loaded_model = pickle.load(open('Random_Forest.sav', 'rb'))
136
  # prediction = loaded_model.predict(np.array(single_sample))
137