ShelterW's picture
Update README.md
7ac60c8
|
raw
history blame
1.16 kB
metadata
datasets:
  - seamew/ChnSentiCorp
language:
  - zh
metrics:
  - accuracy
  - precision
  - f1
  - recall
pipeline_tag: text-classification

hfl-rbt6-ChnSentiCorp-sentiment-classifier

This model is a fine-tuned version of hfl/rbt6 on the seamew/ChnSentiCorp dataset. It achieves the following results on the evaluation set:

  • Loss: 0.294600
  • Accuracy: 0.933884

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • weight_decay=1e-2
  • num_epochs: 3

Training results

Epoch Training Loss Validation Loss Accuracy F1 Precision Recall
1 0.359700 0.306089 0.924242 0.926230 0.918699 0.933884
2 0.200600 0.295512 0.942761 0.943615 0.946755 0.940496
3 0.105600 0.294600 0.933884 0.942452 0.951178 0.933884

Framework versions

  • Pytorch 2.0.0
  • Python 3.9.12