语言:

• 英文 标签:
• 情感分析
• 文本分类 数据集:
• sst2 模型类型: distilbert 许可证: apache-2.0

我的第一个模型

这是一个基于SST-2数据集微调的预训练DistilBERT模型，用于情感分析。它可以预测给定文本表达的是积极还是消极情感。

使用方法

from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("your_username/my-first-model")
tokenizer = AutoTokenizer.from_pretrained("your_username/my-first-model")
inputs = tokenizer("这部电影很棒！", return_tensors="pt")
outputs = model(**inputs)