Downloads · 30 days
7
22% of all-time downloads
twitter_suicide_detection_model
Anshul2000s/twitter_suicide_detection_model
twitter_suicide_detection_model is a text classification model from Anshul2000s. Use it when you need a label for a piece of text. It is set up for transformers.
BERT model fine-tuned to determine whether Twitter posts are potentially suicidal.
All-time downloads
32
Public
Repo size
867 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin433 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jan 7, 2024
- Updated
- Feb 19, 2024
- SHA
- f18b239a
Try a prompt
Model card
From the Hugging Face model README
twitter_suicide_detection_model
BERT model fine-tuned to determine whether Twitter posts are potentially suicidal.
I used Huggingface libraries to perform the fine-tuning.
Format of prompt
prompt = "Is this post potentially suicidal: " + tweet
Before Use
Use the above format to generate the most accurate completions.
Completion
0 = " Not a suicidal post ", 1 = " Potentially Suicidal "
Training/Test split and other information
80:20 % Epochs = 1
I want to be able to run this on a GPU and acquire more training data as well.
This model is in the hugging face repository:
https://huggingface.co/Anshul2000s/twitter_suicide_detection_model
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Created
- Jan 7, 2024
- Updated
- Feb 19, 2024