Back To Projects

Training a Machine Learning Model to Recognize Signs of Depression

Bwohan W.

Many people have not had a professional diagnosis, and could have depression without being aware of it. To attempt to answer this question, I used a person’s daily habits and how they change, and attempt to predict if they will experience a depressive episode in the future.


Depression is a common mental disorder yet can severely damage your everyday habits and health. It is also difficult to diagnose without a professional assessment, so it can be hard to identify whether or not you have it. However, some everyday habits, and changes in those habits, can be indicative of future depressive episodes. Pinpointing those features, I trained a KNN and decision tree based models to predict whether or not someone would have a depressive episode in the future. The best performing model was a KNN model which had an accuracy of 70%, with an F1 score of 69%. These statistics, while not insignificant, still need to be improved on before they can be used as a reliable predictor of depression.

Bwohan W.
Barbie Duckworth
Computer Science BS Alumni MIT, prior Software Engineer at Facebook and eBay

Related Projects

Predicting Dementia Risk with Machine Learning

This research addresses the urgent need for early dementia detection by using machine learning to analyze patient health factors, such as heart rate and diabetes, for dementia prediction, achieving a 52% accuracy with a random forest model and paving the way for future model improvements and deeper insights into influential risk factors.
Min W.
Mentored by John Basbagill
Machine Learning Models for Cardiovascular Disease: A Holistic Review on Early Diagnosis Performance

This paper evaluates the effectiveness of lightweight machine learning models for predicting cardiovascular disease severity, finding high accuracy and potential for practical use, while emphasizing the need for better explainability and integration with wearable technology.
Anirudh C.
Mentored by Matthew Radzihovsky
Cardiac Auscultation: Metrics of Smartphones and Digital Stethoscopes

The objective of our research is to figure out how feasible and accurate a mobile device solution to cardiac ascultation is, compared to a digital stethoscope.
Nicholas T.
Mentored by Sophia Barton