amir7d0/distilbert-base-uncased-finetuned-amazon-reviews

Model Card for distilbert-base-uncased-finetuned-amazon-reviews

Table of Contents

• —Model Card for distilbert-base-uncased-finetuned-amazon-reviews
• —Table of Contents
• —Model Details
• —Uses
• —Fine-tuning hyperparameters
• —Evaluation
• —Framework versions

Model Details

Model Description

<!-- Provide a longer summary of what this model is/does. --> This model is a fine-tuned version of distilbert-base-uncased on amazon_reviews_multi dataset. This model reaches an accuracy of xxx on the dev set.

• —Model type: Language model
• —Language(s) (NLP): en
• —License: apache-2.0
• —Parent Model: For more details about DistilBERT, check out this model card.
• —Resources for more information:
• —Model Documentation

Uses

You can use this model directly with a pipeline for text classification.

from transformers import pipeline

checkpoint = "amir7d0/distilbert-base-uncased-finetuned-amazon-reviews"
classifier = pipeline("text-classification", model=checkpoint)
classifier(["Replace me by any text you'd like."])

and in TensorFlow:

from transformers import AutoTokenizer, TFAutoModelForSequenceClassification

checkpoint = "amir7d0/distilbert-base-uncased-finetuned-amazon-reviews"
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
model = TFAutoModelForSequenceClassification.from_pretrained(checkpoint)

text = "Replace me by any text you'd like."
encoded_input = tokenizer(text, return_tensors='tf')
output = model(encoded_input)

Training Details

Training and Evaluation Data

Here is the raw dataset (amazon_reviews_multi) we used for finetuning the model. The dataset contains 200,000, 5,000, and 5,000 reviews in the training, dev, and test sets respectively.

Fine-tuning hyperparameters

The following hyperparameters were used during training:

• —learning_rate: 2e-05
• —trainbatchsize: 16
• —evalbatchsize: 16
• —seed: 42
• —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• —lrschedulertype: linear
• —num_epochs: 5

Accuracy

The fine-tuned model was evaluated on the test set of amazon_reviews_multi.

• —Accuracy (exact) is the exact match of the number of stars.
• —Accuracy (off-by-1) is the percentage of reviews where the number of stars the model predicts differs by a maximum of 1 from the number given by the human reviewer.

Framework versions

• —Transformers 4.26.1
• —TensorFlow 2.11.0
• —Datasets 2.1.0
• —Tokenizers 0.13.2