Edit model card

Overview

This model was trained with data from https://registry.opendata.aws/helpful-sentences-from-reviews/ to predict how "helpful" a review is.

The model was fine-tuned from the distilbert-base-uncased model

Labels

LABEL_0 - Not helpful
LABEL_1 - Helpful

How to use

The following code shows how to make a prediction with this model

from transformers import (
    AutoTokenizer,
    AutoModelForSequenceClassification,
    TextClassificationPipeline,
)

tokenizer = AutoTokenizer.from_pretrained("banjtheman/distilbert-base-uncased-helpful-amazon")

model = AutoModelForSequenceClassification.from_pretrained(
    "banjtheman/distilbert-base-uncased-helpful-amazon"
)

pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)

result = pipe("This was a Christmas gift for my grandson.")

print(result)

#[{'label': 'LABEL_0', 'score': 0.998775064945221}]
# This is NOT A HELPFUL comment
Downloads last month
33
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.