Twitter-Depression-Detection

Working repository for Group 5 of Wharton Global Youth Summer Program 2024 Data Science Academy.

Notebooks

Project R Notebook

BERT Training

BELT Training

Datasets

Twitter 2015

Harnessing the Power of Hugging Face Transformers for Predicting Mental Health Disorders in Social Networks

Models

2015 BERT

2020 BERT

BELT

For more information see BERT for longer texts

To use this model, clone the above repo and do:

from belt_nlp.bert_classifier_with_pooling import BERTClassifierWithPooling
model = BERTClassifierWithPooling().load(‘path/to/model/directory’)