Downloads · 30 days
0
mishrabp/mini-transformer-sentiment
mini-transformer-sentiment is a text classification model from mishrabp. Use it when you need a label for a piece of text. It is set up for pytorch. The card lists the license as mit.
A minimal Transformer encoder for sentiment classification. Trained on a small custom dataset using PyTorch.
Downloads · 30 days
0
Access
Public
Updated Oct 12, 2025
Repo size
33 MB
Likes
0
Public
Click a slice to open those files.
.pt16.5 MB · 95%
From the Hugging Face model README
A minimal Transformer encoder for sentiment classification.
Trained on a small custom dataset using PyTorch.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
model = AutoModelForSequenceClassification.from_pretrained("mishrabp/mini-transformers")
inputs = tokenizer("I love this!", return_tensors="pt")
print(model(**inputs).logits)