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
- Developed by: Mozilla
- Language(s): English (
en) - License: Apache-2.0
- Fine-tuned from: google/mobilebert-uncased
Model Sources
- Repository: Mozilla Smart Intent Project
- Demo: https://huggingface.co/spaces/chidamnat2002/intent_classifier
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
Created: 2024-10-09
Updated: 2024-12-10
View on Hugging Face