Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -60,8 +60,8 @@ def predict_employee_status(satisfaction_level, last_evaluation, number_project,
|
|
60 |
dmatrix = xgb.DMatrix(input_df)
|
61 |
prediction = model.predict(dmatrix)
|
62 |
prediction_prob = prediction[0]
|
63 |
-
result = "β
Employee is likely to quit." if prediction_prob > 0.
|
64 |
-
return f"{result}
|
65 |
except Exception as e:
|
66 |
return f"β Error: {str(e)}"
|
67 |
|
|
|
60 |
dmatrix = xgb.DMatrix(input_df)
|
61 |
prediction = model.predict(dmatrix)
|
62 |
prediction_prob = prediction[0]
|
63 |
+
result = "β
Employee is likely to quit." if prediction_prob > 0.3 else "β
Employee is likely to stay."
|
64 |
+
return f"{result}"
|
65 |
except Exception as e:
|
66 |
return f"β Error: {str(e)}"
|
67 |
|