Downloads · 30 days
24
19% of all-time downloads
lemonade-sdk/bert-finetuned-phishing-ONNX
bert-finetuned-phishing-ONNX is a text classification model from lemonade-sdk. Use it when you need a label for a piece of text. It is set up for onnx. The card lists the license as apache-2.0.
ONNX export of ealvaradob/bert-finetuned-phishing, packaged for the Lemonade router classifier backend (ort-server).
Downloads · 30 days
24
19% of all-time downloads
All-time downloads
129
Public
Repo size
1.3 GB
Likes
1
Public
Click a slice to open those files.
.onnx1.3 GB · 100%
From the Hugging Face model README
ONNX export of ealvaradob/bert-finetuned-phishing, packaged for the Lemonade router classifier backend (ort-server).
ealvaradob/bert-finetuned-phishingbenign, phishing| file | purpose |
|---|---|
model.onnx | the exported model (input_ids/attention_mask → logits) |
tokenizer.json | the original HuggingFace tokenizer |
manifest.json | task / labels / normalization for ort-server |
export.py | the exact script used to produce & validate these files |
Exported from source with 🤗 Optimum and validated against the original PyTorch model on fixtures (ONNX Runtime CPU vs HF): max softmax delta 0.0 (0 = identical).
CPU-EP latency (ONNX Runtime, single input): ~26.38 ms p50 @ 12 tokens.
pip install "optimum[onnxruntime]" transformers torch onnxruntime sentencepiece
python export.py ealvaradob/bert-finetuned-phishing ./out --task text-classification
See validation.json for the recorded parity result.
Follows the base model ealvaradob/bert-finetuned-phishing; refer to it for terms.