Create README.md

README.md

@@ -0,0 +1,30 @@
+---
+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}")