Our project has developed a tool to estimate how long employees will stay within a company. This tool may be useful for HR analytics and recruitment platforms since it may help predict employees' leave, improving the planning for hiring and keeping employees.
- Open the notebooks on Databricks platform
- (Optional) Run the project - preprocessing notebook, in which the dataset is created
- (Optional) Run the project - features analysis notebook, in which the dataset is imputed
- Run the project - model notebook, in which baselines and xgboost model are trained
- Run the project - NN notebook, in which the FC network is trained