|
|
--- |
|
|
tags: |
|
|
- text-classification |
|
|
library_name: transformers |
|
|
language: en |
|
|
license: apache-2.0 |
|
|
--- |
|
|
|
|
|
# Airline Customer Service Intelligence Model |
|
|
|
|
|
Last Updated: 2025-01-07 22:14:37 |
|
|
Using Checkpoint: checkpoint-1170 |
|
|
|
|
|
## Model Configuration |
|
|
- Base: RoBERTa |
|
|
- Multi-task Classification (Intent, Sentiment, Topic) |
|
|
- Checkpoint: checkpoint-1170 |
|
|
|
|
|
## Usage |
|
|
```python |
|
|
from transformers import AutoModelForSequenceClassification, AutoTokenizer |
|
|
|
|
|
model = AutoModelForSequenceClassification.from_pretrained("raghavdw/cci-capstone") |
|
|
tokenizer = AutoTokenizer.from_pretrained("raghavdw/cci-capstone") |
|
|
|
|
|
text = "I need help with my flight booking" |
|
|
inputs = tokenizer(text, return_tensors="pt") |
|
|
outputs = model(**inputs) |
|
|
``` |
|
|
|
|
|
|
|
|
|
