BERT Base Intent model

This is a fine tuned model based on Bert-Base-Uncased model. This model is used to classify intent into 3 categories- Fintech, Out of Scope and Abusive. The base line model is a pretrained model on English language using a masked language modeling (MLM) objective. It was introduced in this paper and first released in this repository.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-5
  • num_epochs: 3
  • weight_decay:0.01

Training results

Training Loss Epoch Validation Loss Accuracy F1
0.114200 1.0 0.034498 0.991351 0.991346
0.024100 2.0 0.037945 0.992349 0.992355
0.009800 3.0 0.034846 0.993347 0.993345

Model Description

The finetuned Hugging Face model is a variant of the BERT-base-uncased architecture, trained for intent classification with three labels: fintech, abusive, and out of scope. The model has undergone a fine-tuning process, where it has been trained on a large corpus of annotated data using a supervised learning approach. The objective of the model is to classify incoming text data into one of the three predefined classes based on the underlying intent of the text.

The performance of the model was evaluated and it achieved high accuracy and F1 scores for all three classes. The model's high accuracy and robustness make it suitable for use in real-world applications, such as chatbots, customer service automation, and social media monitoring. Overall, the finetuned Hugging Face model provides an effective and reliable solution for intent classification with three labels: fintech, abusive, and out of scope.

  • Developed by: Jeswin MS, Venkatesh R, Kushal S Ballari
  • Model type: Intent Classification
  • Language(s) (NLP): English
  • Finetuned from model: Bert-base-uncased
Downloads last month
12
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Datasets used to train JeswinMS4/bert-base-intent