许可证：apache-2.0
语言：英语
数据集：

• sst2
  评估指标：
• 精确度
• 召回率
• F1值
  标签：
• 文本分类

用于情感分析的GPT-2-medium微调模型 👍👎

基于OpenAI的GPT-2中等模型，在SST-2数据集上微调，用于情感分析下游任务。

GPT-2详情

GPT-2模型由Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya Sutskever在论文语言模型是无监督多任务学习者中提出。

模型微调 🏋️‍

该模型在标准超参数下进行了10个周期的微调。

验证集指标 🧾

模型实战 🚀

from transformers import GPT2Tokenizer, GPT2ForSequenceClassification  

tokenizer = GPT2Tokenizer.from_pretrained("michelecafagna26/gpt2-medium-finetuned-sst2-sentiment")  
model = GPT2ForSequenceClassification.from_pretrained("michelecafagna26/gpt2-medium-finetuned-sst2-sentiment")  

inputs = tokenizer("我喜欢它", return_tensors="pt")  

model(**inputs).logits.argmax(axis=1)  

# 1: 正面, 0: 负面  
# 输出: tensor([1])  

此模型卡片基于Manuel Romero/@mrm8488的"mrm8488/t5-base-finetuned-imdb-sentiment"制作。