Beta
Explore
Marketplace
Neural Labs
Chat
Wallet
Docs
mental-roberta_suicide_base – AI Model by jordanfan | AlphaNeural AI
You can deploy this model and start earning money today!
jordanfan
/
mental-roberta_suicide_base
like
0
peft
safetensors
generated_from_trainer
mental/mental-roberta-base
adapter
cc-by-nc-4.0
us
Views
No views yet
Model card
Files and Versions
Community
API
Deploy
mental-roberta_suicide_base
This model is a fine-tuned version of
mental/mental-roberta-base
on an unknown dataset. It achieves the following results on the evaluation set:
Loss: 0.3474
Accuracy: 0.8532
Precision: 0.8541
Recall: 0.8532
F1: 0.8535
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
learning_rate: 2e-05
train_batch_size: 25
eval_batch_size: 25
seed: 42
optimizer: Use OptimizerNames.ADAMW_8BIT with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
lr_scheduler_type: linear
num_epochs: 1
mixed_precision_training: Native AMP
Training results
Training Loss
Epoch
Step
Validation Loss
Accuracy
Precision
Recall
F1
0.349
1.0
8911
0.3474
0.8532
0.8541
0.8532
0.8535
Framework versions
PEFT 0.14.0
Transformers 4.48.2
Pytorch 2.5.1+cu124
Datasets 3.2.0
Tokenizers 0.21.0
