Setosm/bert-finetuned-for-medical-ner

By Setosm · ↓ 12 · ♥ 0 · Updated 2025-05-16

What's in this repo

This page lists the files in Setosm/bert-finetuned-for-medical-ner, including model.onnx, model_bnb4.onnx, model_fp16.onnx and more.

8 ONNX models (.onnx files)
62 files, including 8 .onnx, 6 .0, 5 .json, 4 .1, 4 .2, 4 .3
Last updated 2025-05-16

ONNX files

File	Size	Download
onnx/model.onnx	—	Download
onnx/model_bnb4.onnx	—	Download
onnx/model_fp16.onnx	—	Download
onnx/model_int8.onnx	—	Download
onnx/model_q4.onnx	—	Download
onnx/model_q4f16.onnx	—	Download
onnx/model_quantized.onnx	—	Download
onnx/model_uint8.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

bert-finetuned-for-medical-ner

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

Loss: 0.1344
Accuracy: 0.9402
Precision: 0.8342
Recall: 0.9029
F1: 0.8672

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: 2
seed: 42
optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
lr_scheduler_type: linear
lr_scheduler_warmup_steps: 2000
num_epochs: 1
mixed_precision_training: Native AMP

Training results

Training Loss	Epoch	Step	Validation Loss	Accuracy	Precision	Recall	F1
0.2039	0.1420	2000	0.1873	0.9244	0.7841	0.8773	0.8281
0.1758	0.2841	4000	0.1613	0.9302	0.8028	0.8920	0.8451
0.1585	0.4261	6000	0.1499	0.9343	0.8309	0.8721	0.8510
0.1613	0.5681	8000	0.1460	0.9358	0.8400	0.8655	0.8526
0.1526	0.7101	10000	0.1402	0.9382	0.8329	0.8930	0.8619
0.1502	0.8522	12000	0.1357	0.9394	0.8346	0.8985	0.8654
0.1486	0.9942	14000	0.1344	0.9402	0.8342	0.9029	0.8672

Framework versions

Transformers 4.51.3
Pytorch 2.6.0+cu124
Datasets 3.6.0
Tokenizers 0.21.1

Model info

token-classification transformers

transformers tensorboard onnx safetensors bert token-classification medical-ner generated_from_trainer en base_model:google-bert/bert-base-uncased base_model:quantized:google-bert/bert-base-uncased license:apache-2.0

↓ 12 ♥ 0

Created: 2025-05-11

Updated: 2025-05-16

View on Hugging Face

Files (62)

.gitattributes

README.md

config.json

model.safetensors

onnx/model.onnx ONNX

onnx/model_bnb4.onnx ONNX

onnx/model_fp16.onnx ONNX

onnx/model_int8.onnx ONNX

onnx/model_q4.onnx ONNX

onnx/model_q4f16.onnx ONNX

onnx/model_quantized.onnx ONNX

onnx/model_uint8.onnx ONNX

quantize_config.json

runs/May11_23-53-10_23084125bf19/events.out.tfevents.1747007599.23084125bf19.1195.0

runs/May11_23-59-27_23084125bf19/events.out.tfevents.1747007970.23084125bf19.1195.1

runs/May12_00-04-48_23084125bf19/events.out.tfevents.1747008291.23084125bf19.1195.2

runs/May12_00-05-13_23084125bf19/events.out.tfevents.1747008316.23084125bf19.1195.3

runs/May12_00-05-47_23084125bf19/events.out.tfevents.1747008349.23084125bf19.1195.4

runs/May12_00-15-12_23084125bf19/events.out.tfevents.1747008915.23084125bf19.1195.5

runs/May12_00-57-36_23084125bf19/events.out.tfevents.1747011458.23084125bf19.1195.6

runs/May12_00-57-48_23084125bf19/events.out.tfevents.1747011471.23084125bf19.1195.7

runs/May12_01-18-04_23084125bf19/events.out.tfevents.1747012688.23084125bf19.1195.8

runs/May12_01-26-29_23084125bf19/events.out.tfevents.1747013192.23084125bf19.1195.9

runs/May12_01-26-51_23084125bf19/events.out.tfevents.1747013214.23084125bf19.1195.10

runs/May12_01-40-29_23084125bf19/events.out.tfevents.1747014031.23084125bf19.1195.11

runs/May12_01-40-29_23084125bf19/events.out.tfevents.1747014259.23084125bf19.1195.12

runs/May12_01-51-02_23084125bf19/events.out.tfevents.1747014666.23084125bf19.1195.13

runs/May12_10-23-35_5b5c14e23ea8/events.out.tfevents.1747045432.5b5c14e23ea8.494.0

runs/May12_10-23-35_5b5c14e23ea8/events.out.tfevents.1747045691.5b5c14e23ea8.494.1

runs/May12_10-23-35_5b5c14e23ea8/events.out.tfevents.1747050284.5b5c14e23ea8.494.2

runs/May12_12-07-54_5b5c14e23ea8/events.out.tfevents.1747051677.5b5c14e23ea8.494.3

runs/May13_19-26-58_d85e24c03750/events.out.tfevents.1747164424.d85e24c03750.1051.0

runs/May13_22-36-56_64c68a430fc5/events.out.tfevents.1747176287.64c68a430fc5.293.0

runs/May13_23-08-49_64c68a430fc5/events.out.tfevents.1747177733.64c68a430fc5.293.1

runs/May13_23-12-08_64c68a430fc5/events.out.tfevents.1747177932.64c68a430fc5.293.2

runs/May13_23-16-52_64c68a430fc5/events.out.tfevents.1747178218.64c68a430fc5.293.3

runs/May13_23-18-35_64c68a430fc5/events.out.tfevents.1747178320.64c68a430fc5.293.4

runs/May13_23-23-00_64c68a430fc5/events.out.tfevents.1747178583.64c68a430fc5.293.5

runs/May13_23-24-22_64c68a430fc5/events.out.tfevents.1747178665.64c68a430fc5.293.6

runs/May13_23-26-34_64c68a430fc5/events.out.tfevents.1747178798.64c68a430fc5.293.7

runs/May13_23-33-17_64c68a430fc5/events.out.tfevents.1747179203.64c68a430fc5.293.8

runs/May13_23-42-26_64c68a430fc5/events.out.tfevents.1747179751.64c68a430fc5.293.9

runs/May13_23-50-12_64c68a430fc5/events.out.tfevents.1747180217.64c68a430fc5.293.10

runs/May13_23-54-25_64c68a430fc5/events.out.tfevents.1747180471.64c68a430fc5.293.11

runs/May13_23-57-13_64c68a430fc5/events.out.tfevents.1747180636.64c68a430fc5.293.12

runs/May14_00-08-18_64c68a430fc5/events.out.tfevents.1747181301.64c68a430fc5.293.13

runs/May14_00-13-08_64c68a430fc5/events.out.tfevents.1747181591.64c68a430fc5.293.14

runs/May14_00-15-49_64c68a430fc5/events.out.tfevents.1747181751.64c68a430fc5.293.15

runs/May14_00-29-13_64c68a430fc5/events.out.tfevents.1747182554.64c68a430fc5.293.16

runs/May15_10-43-33_a96891930ccd/events.out.tfevents.1747305876.a96891930ccd.1024.0

runs/May16_12-39-25_a09b7ee4e4ab/events.out.tfevents.1747399219.a09b7ee4e4ab.659.0

runs/May16_12-53-38_a09b7ee4e4ab/events.out.tfevents.1747400021.a09b7ee4e4ab.659.1

runs/May16_12-56-46_a09b7ee4e4ab/events.out.tfevents.1747400210.a09b7ee4e4ab.659.2

runs/May16_12-59-06_a09b7ee4e4ab/events.out.tfevents.1747400350.a09b7ee4e4ab.659.3

runs/May16_13-00-10_a09b7ee4e4ab/events.out.tfevents.1747400415.a09b7ee4e4ab.659.4

runs/May16_13-39-18_a09b7ee4e4ab/events.out.tfevents.1747402761.a09b7ee4e4ab.659.5

runs/May16_14-26-37_a09b7ee4e4ab/events.out.tfevents.1747405599.a09b7ee4e4ab.659.6

special_tokens_map.json

tokenizer.json

tokenizer_config.json

training_args.bin

vocab.txt

Related models

More from Setosm

No other models by this author.

More token-classification models

Browse all token-classification models

Setosm/bert-finetuned-for-medical-ner

What's in this repo

ONNX files

How to use

README

bert-finetuned-for-medical-ner

Model description

Intended uses & limitations

Training and evaluation data

Training procedure

Training hyperparameters

Training results

Framework versions

Model info

Files (62)

Related models

More from Setosm

More token-classification models

Popular models