louistzx/kaypohaunty

Review Classification Model

Multi-label classification model for restaurant reviews.

Labels

• Advertisement
• Irrelevant Content
• Rant Without Visit
• Spam
• Useful Review

Usage

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

tokenizer = AutoTokenizer.from_pretrained("louistzx/kaypohaunty")
model = AutoModelForSequenceClassification.from_pretrained("louistzx/kaypohaunty")

inputs = tokenizer("Your text here", return_tensors="pt", truncation=True, max_length=256)
with torch.no_grad():
    outputs = model(**inputs)
    predictions = torch.sigmoid(outputs.logits)