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

Stroke Susceptibility Prediction Using Artificial Neural Networks

Our research question is centered around predicting stroke susceptibility and determining which demographic and medical factors significantly increase or decrease that risk. Our results indicate the potentiality of the use of AI in determining stroke risk based solely on collected medical and lifestyle data.
Vinati P.
Mentored by Samuel Kwong
Leveraging AI to Analyze Factors Relating to Social Anxiety

Our research aims to leverage ML to understand what factors play the most critical role in the presence of social anxiety disorder in a person.
Neha K.
Mentored by Udgam Goyal
Modeling Sleep Efficiency

This study explores how alcohol use, exercise, and smoking affect sleep quality by analyzing a dataset containing various lifestyle and sleep-related factors. Using a regression-based predictive model, the research identifies relationships between these behaviors and sleep outcomes, with the final linear regression model achieving a mean absolute error of 0.051.
Nimeesha K.
Mentored by Serena Hughes