Distilbert Base Uncased Helpful Amazon

🚀 レビュー有用性予測モデル

このモデルは、レビューの「有用性」を予測するために、https://registry.opendata.aws/helpful-sentences-from-reviews/ のデータを使用して学習されました。また、distilbert-base-uncased モデルをファインチューニングして作成されています。

🚀 クイックスタート

このモデルは、レビューの有用性を予測するために、特定のデータセットを用いて学習されています。以下に、モデルの基本情報と使用方法を説明します。

✨ 主な機能

• レビューの有用性を予測することができます。
• 予測結果は、"Not helpful" または "Helpful" の2種類で出力されます。

💻 使用例

基本的な使用法

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

ラベル情報

📄 ライセンス

このモデルは、Apache-2.0ライセンスの下で提供されています。