Building a Mental Health Prediction System with Machine Learning, SHAP & Flask

Описание и применение

Создан веб‑приложение для оценки риска психического здоровья, использующее AdaBoost, PHQ‑9, SHAP и Flask, с Docker‑деплоем и мониторингом.

Задача

Develop an end‑to‑end mental‑health assessment tool that combines ML risk prediction, PHQ‑9 screening, explainability, and a web interface, moving beyond a notebook experiment.

Результат

The project delivers a deployable mental‑health risk assessment web app with explainable predictions, PHQ‑9 integration, and support features, demonstrating a complete ML lifecycle from data to production.

Рабочий процесс

• Collect and preprocess PHQ‑9 data
• Train multiple classifiers (AdaBoost, Random Forest, XGBoost, Logistic Regression)
• Select AdaBoost (≈86.9% accuracy)
• Serialize model, transformer, and encoder
• Build Flask backend with Gunicorn
• Create HTML/CSS/Bootstrap frontend with Jinja2 templates
• Integrate SHAP to generate feature‑contribution plots
• Add counseling booking and crisis assistance pages
• Containerize app with Docker
• Deploy to Render via GitHub Actions
• Set up automated uptime monitoring

Доказательства

After tuning and evaluation, the AdaBoost classifier achieved approximately 86.9% accuracy and was selected as the final model.

Источник

The application generates visual explanations showing how individual features contributed to a particular prediction.

Источник

The project also contains serialized ML artifacts such as the trained model, preprocessing transformer and label encoder.

Источник