Update app.py
Browse files
app.py
CHANGED
@@ -84,7 +84,7 @@ data_final = preprocess.transform(data)
|
|
84 |
|
85 |
if st.button('Predict'):
|
86 |
prediction = model.predict(data_final).tolist()[0]
|
87 |
-
prediction = np.where(prediction >= 0.5, 1, 0)
|
88 |
|
89 |
if prediction == 0:
|
90 |
prediction = 'Congratulations, this person most likely will stay!'
|
|
|
84 |
|
85 |
if st.button('Predict'):
|
86 |
prediction = model.predict(data_final).tolist()[0]
|
87 |
+
# prediction = np.where(prediction >= 0.5, 1, 0)
|
88 |
|
89 |
if prediction == 0:
|
90 |
prediction = 'Congratulations, this person most likely will stay!'
|