---
datasets:
- nhull/tripadvisor-split-dataset-v2
base_model:
- huawei-noah/TinyBERT_General_4L_312D
---
# TinyBERT Sentiment Analysis Model

This is a fine-tuned TinyBERT model for sentiment analysis on the Tripadvisor dataset.

## Model Details
- **Base Model**: `huawei-noah/TinyBERT_General_4L_312D`
- **Dataset**: `nhull/tripadvisor-split-dataset-v2`
- **Task**: Multiclass sentiment analysis (5 classes)

## Usage

```python
from transformers import AutoTokenizer, AutoModelForSequenceClassification

# Load the model
tokenizer = AutoTokenizer.from_pretrained("username/tinybert-sentiment-analysis")
model = AutoModelForSequenceClassification.from_pretrained("username/tinybert-sentiment-analysis")

# Predict sentiment
text = "The hotel was amazing and had great service!"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
predicted_class = outputs.logits.argmax().item()
print(f"Predicted class: {predicted_class}")