chidamnat2002's picture
Update README.md
1ac8d2f verified
|
raw
history blame
1.63 kB
metadata
library_name: transformers
license: apache-2.0
language:
  - en
pipeline_tag: text-classification

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} }