Suicide is a serious public health problem. The World Health Organization (WHO) estimates that every year close to 800 000 people take their own life, which is one person every 40 seconds and there are many more people who attempt suicide. Suicide occurs throughout the lifespan and was the second leading cause of death among 25-35 year olds globally in 2016. Hence it is quite clear that suicide is a cause for global concern, so it should be analyzed which are the contributing factors for suicide. Here we have analyzed and visualized the factors affecting the suicide rate in a certain region or country.
The objective of this project is to predict the suicide rates using Machine Learning algorithms and to analyzing significant patterns features that result in increase of suicide rates globally. To see project click here.
The Code is written in Python 3.6.10. If you don't have Python installed you can find it here. If you are using a lower version of Python you can upgrade using the pip package, ensuring you have the latest version of pip. To install the required packages and libraries, run this command in the project directory after cloning the repository:
pip install -r requirements.txt├── pickle
│ ├── label.pkl
│ ├── model.pkl
│ ├── robust.pkl
├── templates
│ ├── index.html
│ ├── styles.css
├── Procfile
├── README.md
├── Suicide Rate Prediction.ipynb
├── app.py
├── requirements.txt
├── suicide_data.csv
Accuracy of various model used for prediction
| ML Model | Train Accuracy | Test Accuracy | Train RMSE | Test RMSE | |
|---|---|---|---|---|---|
| 0 | Bagging Regression | 0.999 | 0.997 | 0.240 | 0.478 |
| 1 | Random Forest | 0.991 | 0.988 | 0.903 | 1.041 |
| 2 | Gradient Boosted Regression | 0.991 | 0.988 | 0.925 | 1.031 |
| 3 | Decision Tree | 0.976 | 0.969 | 1.498 | 1.653 |
| 4 | Multilayer Perceptron Regression | 0.894 | 0.902 | 3.144 | 2.935 |
| 5 | k-Nearest Neighbors Regression | 1.000 | 0.732 | 0.000 | 4.862 |
| 6 | Linear Regression | 0.284 | 0.274 | 8.179 | 8.006 |
| 7 | Support Vector Regression | 0.212 | 0.207 | 8.579 | 8.367 |
Accuracy of Bagging Regression by changing n_estimators value
The final take away form this project is the working of different machine learning models on a dataset and understanding their parameters. Creating this notebook helped me to learn a lot about the parameters of the models, how to tuned them and how they affect the model performance. The final conclusion on the suicide dataset are that the irrespective of age group and generation, male population are more prone to commit suicide than female.