tarekziade/pardonmyai-tiny

By tarekziade · ↓ 7 · ♥ 3 · Updated 2024-03-16

What's in this repo

This page lists the files in tarekziade/pardonmyai-tiny, including model.onnx, model_quantized.onnx.

2 ONNX models (.onnx files)
11 files, including 5 .json, 2 .onnx, 1 .md, 1 .safetensors, 1 .txt
Last updated 2024-03-16

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

Fine-tuned model that detects profanity in text.

This is the "tiny" version, see https://huggingface.co/tarekziade/pardonmyai

Evaluation:

Accuracy: 0.9633
Precision: 0.9378
Recall: 0.8746
F1 Score: 0.9051
AUC-ROC: 0.9892

Model info

text-classification transformers

transformers onnx safetensors bert text-classification dataset:tarekziade/profanity base_model:huawei-noah/TinyBERT_General_4L_312D base_model:quantized:huawei-noah/TinyBERT_General_4L_312D license:apache-2.0 text-embeddings-inference endpoints_compatible region:us

↓ 7 ♥ 3

Created: 2024-03-15

Updated: 2024-03-16

View on Hugging Face

Files (11)

.gitattributes

README.md

config.json

model.safetensors

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

More text-classification models

Browse all text-classification models