jasmeeetsingh's picture
Update README.md
9a8514f
metadata
datasets:
  - sentiment140
metrics:
  - f1
license: apache-2.0
language:
  - en
pipeline_tag: text-classification
tags:
  - twitter
  - depression
  - sentiment140

Model Card for Model ID

jasmeeetsingh/twitter-depression-classification-sentiment140 is a deep learning model trained to classify whether a given tweet is suicidal or not. The model is based on a transformer architecture and fine-tuned on a large corpus of tweets annotated as suicidal or non-suicidal.

Model Details

Model Description

  • Developed by: Jasmeet Singh Sandhu
  • Finetuned from model: paulagarciaserrano/roberta-depression-detection

Uses

The model is intended to be used to classify tweets automatically as suicidal or non-suicidal. It can be used to analyze large volumes of tweets and identify users who may be at risk of depression, as well as to monitor the prevalence of depression-related discussions on social media platforms.

Metrics

Technical Specifications

The model was trained on a 6GB RTX 3060