Distilbert Base Uncased Helpful Amazon

🚀 評論有用性預測模型

本模型旨在預測評論的“有用性”，通過使用來自 https://registry.opendata.aws/helpful-sentences-from-reviews/ 的數據進行訓練。它基於 distilbert-base-uncased 模型進行微調。

🚀 快速開始

模型概述

本模型使用來自 https://registry.opendata.aws/helpful-sentences-from-reviews/ 的數據進行訓練，用於預測評論的“有用性”。它是在 distilbert-base-uncased 模型的基礎上進行微調得到的。

標籤說明

• LABEL_0 - 無用評論
• LABEL_1 - 有用評論

使用方法

以下代碼展示瞭如何使用該模型進行預測：

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 許可證。