ONNX Model Zoo

codewithkyrian/bert-base-uncased-rotten-tomatoes

By codewithkyrian · ↓ 1 · ♥ 0 · Updated 2024-03-14

What's in this repo

This page lists the files in codewithkyrian/bert-base-uncased-rotten-tomatoes, including model.onnx, model_quantized.onnx.

  • 2 ONNX models (.onnx files)
  • 10 files, including 5 .json, 2 .onnx, 1 .md, 1 .txt
  • Last updated 2024-03-14

ONNX files

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

https://huggingface.co/textattack/bert-base-uncased-rotten-tomatoes with ONNX weights to be compatible with Transformers PHP

TextAttack Model Card

This `bert-base-uncased` model was fine-tuned for sequence classificationusing TextAttack 
and the rotten_tomatoes dataset loaded using the `nlp` library. The model was fine-tuned 
for 10 epochs with a batch size of 16, a learning 
rate of 2e-05, and a maximum sequence length of 128. 
Since this was a classification task, the model was trained with a cross-entropy loss function. 
The best score the model achieved on this task was 0.875234521575985, as measured by the 
eval set accuracy, found after 4 epochs.

For more information, check out [TextAttack on Github](https://github.com/QData/TextAttack).

Note: Having a separate repo for ONNX weights is intended to be a temporary solution until WebML gains more traction. If you would like to make your models web-ready, we recommend converting to ONNX using 🤗 Optimum and structuring your repo like this one (with ONNX weights located in a subfolder named onnx).

Model info

text-classification Transformers PHP
Transformers PHP onnx bert text-classification region:us
↓ 1 ♥ 0

Created: 2024-03-14

Updated: 2024-03-14

View on Hugging Face

Files (10)

.gitattributes
README.md
config.json
onnx/model.onnx ONNX
onnx/model_quantized.onnx ONNX
quantize_config.json
special_tokens_map.json
tokenizer.json
tokenizer_config.json
vocab.txt

Related models