This is prediction for Suicide and Non-Suicide: Label-1 is Suicide and Label-0 is Non-Suicide.

Transformers_Project

• Loss: 0.1389
• Accuracy: 0.9672
• F1: 0.9672
• Precision: 0.9676
• Recall: 0.9667
• Zero One Loss: 0.0328

Model description

Intended uses & limitations

Training and evaluation data

Training procedure

Training hyperparameters

• learning_rate: 2e-05
• train_batch_size: 16
• eval_batch_size: 64
• seed: 42
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• num_epochs: 3

Training results

Framework versions

• Transformers 4.38.2
• Pytorch 2.2.1+cu121
• Datasets 2.18.0
• Tokenizers 0.15.2