Model Card for Model ID
Fine tuned model to classify customer support request
Model Details
Model Description
- Model type: ALBERT
- Language(s) (NLP): English
- License: apache-2.0
- Finetuned from model : https://huggingface.co/vineetsharma/customer-support-intent-albert
Uses
How to Get Started with the Model
Use the code below to get started with the model.
[More Information Needed]
Training Details
Training Data
https://huggingface.co/datasets/bitext/Bitext-customer-support-llm-chatbot-training-dataset
Training Procedure
https://colab.research.google.com/drive/1RllMjMO_41l0pf0di1guQePpl97Z6M9A#scrollTo=SVjaVFshS-ox
Preprocessing [optional]
Dividing dataset 80/20% for finetuning and evaluation
Training Hyperparameters
- evaluation_strategy:epoch
- learning_rate: 2e-5
- per_device_train_batch_size: 16
- per_device_eval_batch_size: 16
- num_train_epochs: 3
- weight_decay: 0.01
Metrics & Evaluation
- eval_loss: 0.002536403015255928,
- eval_accuracy: 0.9996279069767442
- eval_runtime: 11.3331
- eval_samples_per_second: 474.275
- eval_steps_per_second: 29.648
- epoch: 3.0
Testing Data, Factors & Metrics
Testing Data
20% of "https://huggingface.co/vineetsharma/customer-support-intent-albert" dataset
accuracy
- Downloads last month
- 0
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
the model is not deployed on the HF Inference API.
Model tree for AIEnthusiast369/customer-support-categ_classification-albert_v2
Base model
albert/albert-base-v2