bert-base-uncased-finetuned-ner

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

  • Loss: 0.9799
  • Overall Precision: 0.4362
  • Overall Recall: 0.5285
  • Overall F1: 0.4779
  • Overall Accuracy: 0.9515
  • Datasetname F1: 0.3262
  • Hyperparametername F1: 0.7339
  • Hyperparametervalue F1: 0.7619
  • Methodname F1: 0.5193
  • Metricname F1: 0.6525
  • Metricvalue F1: 0.6452
  • Taskname F1: 0.2704

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: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 100

Training results

Training Loss Epoch Step Validation Loss Overall Precision Overall Recall Overall F1 Overall Accuracy Datasetname F1 Hyperparametername F1 Hyperparametervalue F1 Methodname F1 Metricname F1 Metricvalue F1 Taskname F1
No log 1.0 132 0.5733 0.2215 0.3313 0.2655 0.9350 0.0 0.1273 0.325 0.4369 0.2759 0.3478 0.0639
No log 2.0 264 0.4130 0.2929 0.5996 0.3936 0.9348 0.2930 0.3316 0.6133 0.5104 0.5622 0.5263 0.2452
No log 3.0 396 0.4654 0.3302 0.5793 0.4207 0.9431 0.2732 0.5 0.7463 0.5189 0.5930 0.5405 0.2228
0.5901 4.0 528 0.4540 0.3604 0.5772 0.4438 0.9457 0.2644 0.5271 0.7027 0.5346 0.6386 0.4878 0.2804
0.5901 5.0 660 0.5776 0.3920 0.5346 0.4523 0.9483 0.2529 0.6241 0.7761 0.5056 0.6369 0.6875 0.2601
0.5901 6.0 792 0.4903 0.3934 0.6037 0.4763 0.9493 0.3651 0.6230 0.6667 0.5775 0.6136 0.4848 0.3084
0.5901 7.0 924 0.6343 0.3991 0.5549 0.4643 0.9492 0.3286 0.6047 0.6301 0.5705 0.6216 0.5 0.2508
0.1282 8.0 1056 0.6686 0.4185 0.5691 0.4823 0.9507 0.3546 0.6565 0.7353 0.5786 0.6174 0.4615 0.2540
0.1282 9.0 1188 0.6801 0.4144 0.5610 0.4767 0.9507 0.3729 0.5616 0.7576 0.5569 0.5839 0.5263 0.2821
0.1282 10.0 1320 0.5843 0.4022 0.5976 0.4808 0.9493 0.3262 0.6667 0.6957 0.5417 0.6267 0.6286 0.2969
0.1282 11.0 1452 0.7041 0.4187 0.5813 0.4868 0.9508 0.3158 0.608 0.6857 0.5583 0.6174 0.7429 0.3145
0.0579 12.0 1584 0.7037 0.4143 0.5894 0.4866 0.9509 0.3310 0.6032 0.7164 0.5426 0.6497 0.6486 0.3207
0.0579 13.0 1716 0.6623 0.3961 0.5732 0.4684 0.9512 0.3490 0.6847 0.6462 0.5411 0.5987 0.6286 0.3038
0.0579 14.0 1848 0.7953 0.4443 0.5752 0.5013 0.9517 0.3704 0.6457 0.6667 0.5705 0.625 0.6667 0.3030
0.0579 15.0 1980 0.7321 0.4050 0.5976 0.4828 0.9482 0.3023 0.7156 0.7042 0.5815 0.6309 0.6667 0.2935
0.0281 16.0 2112 0.7175 0.4149 0.6240 0.4984 0.9503 0.3740 0.5816 0.6667 0.5756 0.6335 0.6667 0.3182
0.0281 17.0 2244 0.7463 0.4242 0.5915 0.4941 0.9507 0.3741 0.7130 0.7429 0.5678 0.6144 0.6486 0.2824
0.0281 18.0 2376 0.8040 0.4172 0.5732 0.4829 0.9531 0.3485 0.6838 0.7647 0.4946 0.6279 0.6286 0.3233
0.0156 19.0 2508 0.8922 0.4365 0.5589 0.4902 0.9525 0.4 0.6607 0.7500 0.5455 0.6013 0.6316 0.2803
0.0156 20.0 2640 0.8597 0.4072 0.5488 0.4675 0.9516 0.3333 0.6435 0.7213 0.5086 0.6536 0.5882 0.2975
0.0156 21.0 2772 0.7927 0.3986 0.5752 0.4709 0.9518 0.3759 0.6107 0.6857 0.5048 0.6582 0.7273 0.2802
0.0156 22.0 2904 0.9151 0.4234 0.5732 0.4870 0.9493 0.3590 0.7143 0.8060 0.5491 0.6174 0.7273 0.2305
0.0102 23.0 3036 0.9133 0.4231 0.5813 0.4897 0.9502 0.3185 0.6357 0.7941 0.55 0.6795 0.6061 0.2754
0.0102 24.0 3168 0.9799 0.4362 0.5285 0.4779 0.9515 0.3262 0.7339 0.7619 0.5193 0.6525 0.6452 0.2704

Framework versions

  • Transformers 4.23.1
  • Pytorch 1.12.1+cu102
  • Datasets 2.6.1
  • Tokenizers 0.13.1
Downloads last month
14
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.