ONNX Model Zoo

ml-projects/clickbait-ml_bert

By ml-projects · ↓ 0 · ♥ 0 · Updated 2023-06-13

What's in this repo

This page lists the files in ml-projects/clickbait-ml_bert, including model_quantized.onnx.

  • 1 ONNX models (.onnx files)
  • 9 files, including 4 .json, 1 .md, 1 .onnx, 1 .h5, 1 .txt
  • Last updated 2023-06-13

ONNX files

File Size Download
onnx/model_quantized.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

clickbait-ml_bert

This model is a fine-tuned version of bert-base-german-cased on an unknown dataset. It achieves the following results on the evaluation set:

  • Train Loss: 0.6057
  • Validation Loss: 0.6160
  • Train Accuracy: 0.8235
  • Epoch: 3

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:

  • optimizer: {'name': 'Adam', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 8, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False}
  • training_precision: float32

Training results

Train Loss Validation Loss Train Accuracy Epoch
0.7115 0.6299 0.8235 0
0.6071 0.6160 0.8235 1
0.5783 0.6160 0.8235 2
0.6057 0.6160 0.8235 3

Framework versions

  • Transformers 4.30.1
  • TensorFlow 2.12.0
  • Datasets 2.12.0
  • Tokenizers 0.13.3

Model info

text-classification transformers
transformers tf onnx bert text-classification generated_from_keras_callback de dataset:ml-projects/clickbait-ml_dataset license:mit text-embeddings-inference endpoints_compatible region:us
↓ 0 ♥ 0

Created: 2023-06-11

Updated: 2023-06-13

View on Hugging Face

Files (9)

.gitattributes
README.md
config.json
onnx/model_quantized.onnx ONNX
special_tokens_map.json
tf_model.h5
tokenizer.json
tokenizer_config.json
vocab.txt

Related models