Views
No views yet
torch.nn — no pretrained weights.| Parameter | Value |
|---|---|
| Hidden dim | 256 |
| FFN dim | 1024 |
| Layers | 6 |
| Heads | 8 |
| Max seq len | 256 |
| Vocab size | 16000 |
| Labels | NEGATIVE, NEUTRAL, POSITIVE |
| Precision | bf16 mixed-precision |
1from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
2
3model = AutoModelForSequenceClassification.from_pretrained(
4 "Impulse2000/sentiment-transformer", trust_remote_code=True
5)
6tokenizer = AutoTokenizer.from_pretrained("Impulse2000/sentiment-transformer")
7pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
8print(pipe("This movie was absolutely fantastic!"))