Text Classification
Transformers
Safetensors
English
modernbert
intent-classification
chatbot
enterprise-chatbot
nlp
text-embeddings-inference
Instructions to use yogeshpandey586/chatbot-intent-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use yogeshpandey586/chatbot-intent-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="yogeshpandey586/chatbot-intent-classifier")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("yogeshpandey586/chatbot-intent-classifier") model = AutoModelForSequenceClassification.from_pretrained("yogeshpandey586/chatbot-intent-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| language: | |
| - en | |
| license: apache-2.0 | |
| library_name: transformers | |
| pipeline_tag: text-classification | |
| tags: | |
| - modernbert | |
| - intent-classification | |
| - chatbot | |
| - enterprise-chatbot | |
| - nlp | |
| - text-classification | |
| base_model: answerdotai/ModernBERT-base | |
| # ModernBERT Chatbot Intent Classifier | |
| A fine-tuned **ModernBERT** model for enterprise chatbot intent classification. This model classifies user queries into predefined intent categories and is designed to be used as the first stage of a chatbot pipeline before Retrieval-Augmented Generation (RAG) or other downstream workflows. | |
| --- | |
| # Supported Intents | |
| The model predicts one of the following intents: | |
| - Business | |
| - Greeting | |
| - SmallTalk | |
| - Complaint | |
| - Feedback | |
| - Thanks | |
| - Goodbye | |
| - OutOfDomain | |
| --- | |
| # Intended Use | |
| This model is designed for: | |
| - Enterprise AI Chatbots | |
| - Customer Support Bots | |
| - HR Assistants | |
| - Knowledge Base Assistants | |
| - Intent Detection | |
| - Query Routing | |
| - RAG Pipeline Routing | |
| Example flow: | |
| ``` | |
| User Query | |
| β | |
| βΌ | |
| Intent Classifier | |
| β | |
| βββ Business | |
| β βββ RAG Pipeline | |
| β | |
| βββ Greeting | |
| β βββ Greeting Response | |
| β | |
| βββ SmallTalk | |
| β βββ Small Talk Response | |
| β | |
| βββ Complaint | |
| β βββ Complaint Workflow | |
| β | |
| βββ Feedback | |
| β βββ Feedback Workflow | |
| β | |
| βββ Thanks | |
| β βββ Thank You Response | |
| β | |
| βββ Goodbye | |
| β βββ Goodbye Response | |
| β | |
| βββ OutOfDomain | |
| βββ Reject Politely | |
| ``` | |
| --- | |
| # Base Model | |
| - **Model:** answerdotai/ModernBERT-base | |
| - **Task:** Sequence Classification | |
| - **Framework:** Hugging Face Transformers | |
| - **Language:** English | |
| --- | |
| # Example Usage | |
| ```python | |
| from transformers import pipeline | |
| classifier = pipeline( | |
| "text-classification", | |
| model="yogeshpandey586/chatbot-intent-classifier" | |
| ) | |
| print(classifier("Hello")) | |
| print(classifier("What services do you provide?")) | |
| print(classifier("Tell me a joke")) | |
| ``` | |
| Example Output | |
| ```python | |
| [ | |
| { | |
| "label": "Greeting", | |
| "score": 0.998 | |
| } | |
| ] | |
| ``` | |
| --- | |
| # Training Dataset | |
| The model was fine-tuned using a custom intent classification dataset containing manually labeled chatbot queries. | |
| Each sample contains: | |
| - User Query | |
| - Intent Label | |
| Example: | |
| | Query | Intent | | |
| |-------|--------| | |
| | Hello | Greeting | | |
| | Hi | Greeting | | |
| | Good Morning | Greeting | | |
| | What services do you provide? | Business | | |
| | How can I reset my password? | Business | | |
| | Tell me a joke | SmallTalk | | |
| | Thanks | Thanks | | |
| | Bye | Goodbye | | |
| | I have feedback | Feedback | | |
| | Your service is bad | Complaint | | |
| | Who won yesterday's IPL match? | OutOfDomain | | |
| --- | |
| # Training Details | |
| - Base Model: ModernBERT-base | |
| - Fine-tuning Task: Sequence Classification | |
| - Framework: Hugging Face Transformers | |
| - Optimizer: AdamW | |
| - Loss Function: Cross Entropy Loss | |
| --- | |
| # Evaluation | |
| The model was evaluated using a validation dataset. | |
| Evaluation Metric: | |
| - Accuracy | |
| This model is intended for production chatbot routing where low latency and high intent classification accuracy are required. | |
| --- | |
| # Limitations | |
| This model is intended only for intent classification. | |
| It is **not** designed for: | |
| - Toxicity Detection | |
| - Sentiment Analysis | |
| - Emotion Detection | |
| - Translation | |
| - Summarization | |
| - Question Answering | |
| - Text Generation | |
| Performance may decrease on: | |
| - Languages other than English | |
| - Ambiguous user queries | |
| - Multi-intent queries | |
| - Queries outside the training domain | |
| --- | |
| # Production Architecture | |
| ``` | |
| User Query | |
| β | |
| βΌ | |
| ModernBERT Intent Classifier | |
| β | |
| βΌ | |
| Intent Router | |
| β | |
| βββ Business β RAG | |
| βββ Greeting β Greeting Service | |
| βββ SmallTalk β Small Talk Service | |
| βββ Complaint β Complaint Service | |
| βββ Feedback β Feedback Service | |
| βββ Thanks β Thank You Service | |
| βββ Goodbye β Goodbye Service | |
| βββ OutOfDomain β Reject Response | |
| ``` | |
| --- | |
| # Technical Specifications | |
| | Property | Value | | |
| |----------|-------| | |
| | Architecture | ModernBERT | | |
| | Base Model | answerdotai/ModernBERT-base | | |
| | Framework | Transformers | | |
| | Task | Sequence Classification | | |
| | Language | English | | |
| | Deployment | CPU / GPU | | |
| --- | |
| # Developer | |
| **Yogesh Pandey** | |
| AI Developer | |
| --- | |
| # License | |
| Apache-2.0 | |
| --- | |
| # Citation | |
| ```bibtex | |
| @misc{yogeshpandey2026intentclassifier, | |
| title={ModernBERT Chatbot Intent Classifier}, | |
| author={Yogesh Pandey}, | |
| year={2026}, | |
| publisher={Hugging Face} | |
| } | |
| ``` |