Model: vibhorag101/roberta-base-suicide-prediction-phr
Author: vibhorag101
Pipeline: text-classification
Library: transformers
Tags: transformers, pytorch, roberta, text-classification, generated_from_trainer, en, dataset:vibhorag101/suicide_prediction_dataset_phr, base_model:FacebookAI/roberta-base, base_model:finetune:FacebookAI/roberta-base, license:mit, model-index, endpoints_compatible, region:us
Base models (finetune): id: FacebookAI/roberta-base
Card metadata: license: mit
base_model: roberta-base
tags: generated_from_trainer
metrics: accuracy, recall, precision, f1
model-index: name: roberta-base-suicide-prediction-phr
results: task: type: text-classification
name: Suicidal Tendency Prediction in text
dataset: type: vibhorag101/roberta-base-suicide-prediction-phr
name: Suicide Prediction Dataset
split: test
metrics: type: accuracy
value: 0.9652972367116438
verified: no, type: f1
value: 0.9651921995935487
verified: no, type: recall
value: 0.966571403827834
verified: no, type: precision
value: 0.9638169257340242
verified: no
datasets: vibhorag101/suicide_prediction_dataset_phr
language: en
library_name: transformers
Likes: 12 | All-time downloads: 1758318
Created: 2023-11-24 17:36:11