ONNX Model Zoo

elihoole/modern-bert-finetuned-query-classification

By elihoole · ↓ 8 · ♥ 0 · Updated 2025-04-30

What's in this repo

This page lists the files in elihoole/modern-bert-finetuned-query-classification, including model.onnx and more.

  • 3 ONNX models (.onnx files)
  • 24 files, including 7 .json, 3 .onnx, 3 .0, 3 .1, 2 .2, 2 .3
  • Last updated 2025-04-30

ONNX files

File Size Download
onnx_version/model.onnx — Download
onnx_version/v_2/model.onnx — Download
onnx_version/v_3/model.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

modern-bert-finetuned-query-classification

This model is a fine-tuned version of answerdotai/ModernBERT-base on an unknown dataset. It achieves the following results on the evaluation set:

  • Loss: 0.1256
  • Accuracy: 0.9759
  • F1: 0.9759
  • Precision: 0.9763
  • Recall: 0.9759

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: 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
  • num_epochs: 5

Training results

Training Loss Epoch Step Validation Loss Accuracy F1 Precision Recall
No log 1.0 315 0.1474 0.9648 0.9647 0.9649 0.9648
0.1965 2.0 630 0.1226 0.9704 0.9704 0.9718 0.9704
0.1965 3.0 945 0.1192 0.9741 0.9742 0.9757 0.9741
0.0426 4.0 1260 0.1250 0.9741 0.9741 0.9742 0.9741
0.0042 5.0 1575 0.1256 0.9759 0.9759 0.9763 0.9759

Framework versions

  • Transformers 4.51.3
  • Pytorch 2.6.0+cu124
  • Datasets 3.5.1
  • Tokenizers 0.21.1

Model info

text-classification transformers
transformers tensorboard onnx safetensors modernbert text-classification generated_from_trainer base_model:answerdotai/ModernBERT-base base_model:quantized:answerdotai/ModernBERT-base license:apache-2.0 text-embeddings-inference endpoints_compatible
↓ 8 ♥ 0

Created: 2025-04-10

Updated: 2025-04-30

View on Hugging Face

Files (24)

.gitattributes
README.md
config.json
model.safetensors
onnx_version/id2label.json
onnx_version/model.onnx ONNX
onnx_version/v_2/id2label.json
onnx_version/v_2/model.onnx ONNX
onnx_version/v_3/id2label.json
onnx_version/v_3/model.onnx ONNX
runs/Apr10_11-32-12_ef403fb20d9e/events.out.tfevents.1744284745.ef403fb20d9e.188.0
runs/Apr10_11-32-12_ef403fb20d9e/events.out.tfevents.1744285331.ef403fb20d9e.188.1
runs/Apr10_11-49-18_ef403fb20d9e/events.out.tfevents.1744285766.ef403fb20d9e.188.2
runs/Apr10_11-49-18_ef403fb20d9e/events.out.tfevents.1744286461.ef403fb20d9e.188.3
runs/Apr22_08-25-04_7e8cfa1e754a/events.out.tfevents.1745310341.7e8cfa1e754a.369.0
runs/Apr22_08-25-04_7e8cfa1e754a/events.out.tfevents.1745311264.7e8cfa1e754a.369.1
runs/Apr30_06-11-19_68be676fb8aa/events.out.tfevents.1745993502.68be676fb8aa.447.0
runs/Apr30_06-11-19_68be676fb8aa/events.out.tfevents.1745994824.68be676fb8aa.447.1
runs/Apr30_06-49-46_68be676fb8aa/events.out.tfevents.1745995800.68be676fb8aa.447.2
runs/Apr30_06-49-46_68be676fb8aa/events.out.tfevents.1745996410.68be676fb8aa.447.3
special_tokens_map.json
tokenizer.json
tokenizer_config.json
training_args.bin

Related models

More from elihoole

No other models by this author.

More text-classification models

Browse all text-classification models

Popular models