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 许可证。