Gpt2 Medium Finetuned Sst2 Sentiment

🚀 GPT-2-medium 情感分析微調模型 👍👎

本項目基於 OpenAI 的 GPT-2 中型模型，在 SST-2 數據集上進行微調，用於情感分析下游任務。

🚀 快速開始

代碼示例

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("I love it", return_tensors="pt")

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

# 1: Positive, 0: Negative
# Output: tensor([1])

✨ 主要特性

• 基於強大的 GPT-2 中型模型進行微調，適用於情感分析任務。
• 在標準超參數下進行了 10 個輪次的微調，具有較好的性能表現。

📚 詳細文檔

GPT-2 詳情

GPT-2 模型由 Alec Radford、Jeffrey Wu、Rewon Child、David Luan、Dario Amodei、Ilya Sutskever 在論文 Language Models are Unsupervised Multitask Learners 中提出。

模型微調 🏋️‍

該模型在標準超參數下進行了 10 個輪次的微調。

驗證集指標 🧾

📄 許可證

本項目採用 Apache-2.0 許可證。

💡 使用提示：本模型卡片基於 Manuel Romero/@mrm8488 的 "mrm8488/t5-base-finetuned-imdb-sentiment"。