MMLForge Search models, datasets, tasks… Start free
Home › Models › Text Classification › vibhorag101/roberta-base-suicide-prediction-phr
R

vibhorag101/roberta-base-suicide-prediction-phr

Text Classification·vibhorag101· 88.5K· 12
transformers mit Text Classification dataset:vibhorag101/suicide_prediction_dataset_phrbase_model:FacebookAI/roberta-basebase_model:finetune:FacebookAI/roberta-baselicense:mit

This model is a fine-tuned version of roberta-base on this dataset sourced from Reddit. It achieves the following results on the evaluation/validation set: - Loss: 0.1543 - Accuracy: 0.9652972367116438 - Recall: 0.966571403827834 - Precision: 0.9638169257340242 - F1: 0.9651921995935487

Open in MLForge Sign up free Desktop app Source ↗
# pull & run locally
pip install mlforge-sdk && mlforge pull vibhorag101/roberta-base-suicide-prediction-phr

Model details

Task
Text Classification
Provider
vibhorag101
Framework
transformers
Size
16 GB
License
mit
Downloads
88.5K
Likes
12
Updated
2025-10-22

About vibhorag101/roberta-base-suicide-prediction-phr

This model is a fine-tuned version of roberta-base on this dataset sourced from Reddit. It achieves the following results on the evaluation/validation set: - Loss: 0.1543 - Accuracy: 0.9652972367116438 - Recall: 0.966571403827834 - Precision: 0.9638169257340242 - F1: 0.9651921995935487

Related Text Classification

B BAAI/bge-reranker-v2-m3 Text Classification ·567.8M params 16.2M 1.1K 🤗 HF F ProsusAI/finbert Text Classification 7.6M 1.2K 🤗 HF B BAAI/bge-reranker-base Text Classification ·278.0M params 4.1M 238 🤗 HF T cardiffnlp/twitter-roberta-base-sentiment-latest Text Classification 3.9M 813 🤗 HF D distilbert/distilbert-base-uncased-finetuned-sst-2-english Text Classification ·67.0M params 3.6M 910 🤗 HF T trl-internal-testing/tiny-Qwen2ForSequenceClassification-2.5 Text Classification ·2.4M params 1.5M 1 🤗 HF

Browse all Text Classification models →