ONNX Model Zoo

Mozilla/mobilebert-uncased-finetuned-LoRA-intent-classifier

By Mozilla · ↓ 6 · ♥ 0 · Updated 2024-12-10

What's in this repo

This page lists the files in Mozilla/mobilebert-uncased-finetuned-LoRA-intent-classifier, including model.onnx, model_fp16.onnx, model_q4.onnx and more.

  • 4 ONNX models (.onnx files)
  • 13 files, including 5 .json, 4 .onnx, 1 .md, 1 .safetensors, 1 .txt
  • Last updated 2024-12-10

ONNX files

File Size Download
onnx/model.onnx — Download
onnx/model_fp16.onnx — Download
onnx/model_q4.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 LoRA Classifier on MobileBERT

This is a fine-tuned LoRA (Low-Rank Adaptation) classifier on MobileBERT, designed for intent classification across multiple categories. MobileBERT is a compact and efficient version of BERT_LARGE, optimized for resource-constrained devices while maintaining robust performance. The model has been fine-tuned to classify intents such as information queries, navigation, purchase, and more.

Model Details

Model Description

This model is based on MobileBERT (uncased_L-24_H-128_B-512_A-4_F-4_OPT) and fine-tuned using LoRA for intent classification. The fine-tuning process adapts the model to predict intents across 8 categories:

  • information_intent
  • yelp_intent
  • navigation_intent
  • travel_intent
  • purchase_intent
  • weather_intent
  • translation_intent
  • unknown

Model Sources

Citation

If you use this model, please cite it as:

@misc{mozilla_mobilebert_lora_intent,
  title       = {Fine-tuned LoRA Classifier on MobileBERT},
  author      = {Mozilla},
  year        = {2024},
  url         = {https://huggingface.co/Mozilla/mobilebert-uncased-finetuned-LoRA-intent-classifier},
  license     = {Apache-2.0}
}

Model info

text-classification transformers
transformers onnx safetensors mobilebert text-classification en license:apache-2.0 endpoints_compatible region:us
↓ 6 ♥ 0

Created: 2024-10-09

Updated: 2024-12-10

View on Hugging Face

Files (13)

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

Related models