Mini Transformer Sentiment Model

A minimal Transformer encoder for sentiment classification.
Trained on a small custom dataset using PyTorch.

Usage

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)
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support