CoolFace
Modelpublic

eunyounglee/hana-persona-emotion-bert-simple-finetuning-1

sourceHugging Facecc-by-sa-4.0updated 3y agoView on Hugging Face
0likes12downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

hana-persona-emotion-bert-simple-finetuning-1

This model is a fine-tuned version of klue/bert-base on the None dataset. It achieves the following results on the evaluation set:

  • —Loss: 1.5295
  • —Accuracy: 0.8043
  • —F1: 0.8041

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
  • —trainbatchsize: 16
  • —evalbatchsize: 16
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 10

Training results

Training LossEpochStepValidation LossAccuracyF1
0.49831.022840.47530.80520.8038
0.37282.045680.49230.80980.8078
0.28273.068520.53510.80700.8074
0.19064.091360.73750.81060.8099
0.1555.0114200.89060.80410.8040
0.10876.0137041.10590.80500.8050
0.07327.0159881.28820.80490.8053
0.06328.0182721.38000.80660.8062
0.03699.0205561.49390.80350.8028
0.015910.0228401.52950.80430.8041

Framework versions

  • —Transformers 4.38.1
  • —Pytorch 2.2.0
  • —Datasets 2.17.1
  • —Tokenizers 0.15.2