How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="alanwang2001/BERT-sentiment-lora")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("alanwang2001/BERT-sentiment-lora")
model = AutoModelForSequenceClassification.from_pretrained("alanwang2001/BERT-sentiment-lora")
Quick Links

BERT-sentiment-lora

This model is a fine-tuned version of google-bert/bert-base-uncased for 3-class movie review sentiment classification.

Labels

  • 0 โ†’ negative (โˆ’1)
  • 1 โ†’ mixed (0)
  • 2 โ†’ positive (+1)

Training details

  • Architecture: BertForSequenceClassification
  • Hidden size: 768, Layers: 12, Heads: 12
  • Framework: Transformers 5.3.0
Downloads last month
1
Safetensors
Model size
0.1B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for alanwang2001/BERT-sentiment-lora

Finetuned
(6799)
this model