tanaos-intent-classifier-v1

<p align="center"> <img src="https://raw.githubusercontent.com/tanaos/.github/master/assets/logo.png" width="250px" alt="Tanaos – Train task specific LLMs without training data, for offline NLP and Text Classification"> </p>

tanaos-intent-classifier-v1: A small but performant intent classification model

This model was created by Tanaos with the Artifex Python library.

This is an intent classification model based on FacebookAI/roberta-base and fine-tuned on a synthetic dataset to classify text into one of 12 different intent categories:

These categories were chosen to cover a wide range of common user intents in chatbot and virtual assistant interactions, regardless of the specific application domain, in order to create a versatile and general-purpose intent classification model, applicable across various industries and use cases.

This model does not have an explicit fallback or unknown category. Instead, it is trained to always classify input text into one of the predefined intent categories, even if the input does not clearly belong to any of them. <ins>We advise users to interpret text inputs leading to predictions with a score field lower than 0.5 as implicitly belonging to the fallback or unknown intent category</ins>.

How to Use

Use this model through the Artifex library:

install Artifex with

pip install artifex

use the model with

from artifex import Artifex

intent_classifier = Artifex().intent_classifier()

label = intent_classifier("Hey there, how are you doing?")
print(label)

# >>> [{'label': 'greeting', 'score': 0.9955}]

Model Description

• Base model: FacebookAI/roberta-base
• Task: Text classification (intent classification)
• Languages: English
• Fine-tuning data: A synthetic, custom dataset of 10,000 utterances, each belonging to one of 12 different intent categories.

Training Details

This model was trained using the Artifex Python library

pip install artifex

by providing the following instructions and generating 10,000 synthetic training samples:

from artifex import Artifex


intent_classifier = Artifex().intent_classifier()

intent_classifier.train(
    classes={
        "greeting": "Intent to greet or say hello.",
        "farewell": "Intent to say goodbye or farewell.",
        "thank_you": "Intent to express gratitude or thanks.",
        "affirmation": "Intent to agree or confirm something.",
        "negation": "Intent to disagree or deny something.",
        "small_talk": "Intent to engage in casual or light conversation with no specific purpose.",
        "bot_capabilities": "Inquiries about the bot's features or abilities.",
        "feedback_positive": "Intent to provide positive feedback about the bot, service, or experience.",
        "feedback_negative": "Intent to provide negative feedback about the bot, service, or experience.",
        "clarification": "Intent to ask for clarification or more information about a previous statement or question.",
        "suggestion": "Intent to offer a suggestion or recommendation for improvement.",
        "language_change": "Intent to request a change in the language being used by the bot or information about language options.",
    },
    num_samples=10000
)

Intended Uses

This model is intended to:

• Classify user intents in chatbot and virtual assistant interactions.
• Be used in various industries and applications, such as customer support, virtual assistants, and conversational agents.
• Assist in routing user queries to appropriate responses or actions based on detected intent.

Not intended for:

• Highly specialized domains where intents differ significantly from the predefined categories.