rasmodev commited on
Commit
826bd0c
Β·
verified Β·
1 Parent(s): 8f83d49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -140,14 +140,14 @@ def main():
140
  # Display predicted attrition status
141
  st.subheader("Predicted Attrition Status 🎯")
142
  if prediction[0] == 1:
143
- st.error("Employee is predicted to leave (Attrition = Yes)")
144
  else:
145
- st.success("Employee is predicted to stay (Attrition = No)")
146
 
147
  # Display prediction probability
148
  if prediction[0] == 1:
149
  st.subheader("Prediction Probability πŸ“ˆ")
150
- st.write(f"The probability of the employee leaving is: {probability[0]*100:.2f}%")
151
 
152
  # Display characteristic-based recommendations
153
  st.subheader("Recommendations for Retaining The Employee πŸ’‘:")
 
140
  # Display predicted attrition status
141
  st.subheader("Predicted Attrition Status 🎯")
142
  if prediction[0] == 1:
143
+ st.error("at higher risk of leaving the organization")
144
  else:
145
+ st.success("Employee is predicted to stay in the organization")
146
 
147
  # Display prediction probability
148
  if prediction[0] == 1:
149
  st.subheader("Prediction Probability πŸ“ˆ")
150
+ st.write(f"The probability of the employee leaving the organization is: {probability[0]*100:.2f}%")
151
 
152
  # Display characteristic-based recommendations
153
  st.subheader("Recommendations for Retaining The Employee πŸ’‘:")