Team Ai
Modelpublic

gyr66/relation_extraction_bert_base_uncased

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
5likes285downloads
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. -->

relationextractionbertbaseuncased

This model is a fine-tuned version of bert-base-uncased on sem_eval_2010_task_8. It achieves the following results on the evaluation set:

  • —Loss: 0.9300
  • —F1: 0.8616

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

Training results

Training LossEpochStepValidation LossF1
1.25111.01070.60980.7907
0.4372.02140.51670.8453
0.19213.03210.54460.8553
0.07954.04280.70410.8527
0.03345.05350.77740.8568
0.01246.06420.80860.8490
0.00567.07490.86150.8601
0.0038.08560.92460.8614
0.00149.09630.93000.8616
0.001110.010700.94210.8613

Framework versions

  • —Transformers 4.35.2
  • —Pytorch 2.1.0+cu121
  • —Datasets 2.16.0
  • —Tokenizers 0.15.0