Gpt2 Medium Finetuned Sst2 Sentiment

🚀 感情分析用にファインチューニングされたGPT-2-medium 👍👎

このモデルは、OpenAIのGPT-2 mediumを、感情分析という下流タスクのために、SST-2データセットでファインチューニングしたものです。

📚 詳細ドキュメント

GPT-2の詳細

GPT-2モデルは、Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya SutskeverによるLanguage Models are Unsupervised Multitask Learnersで発表されました。

モデルのファインチューニング 🏋️‍

このモデルは、標準的なハイパーパラメータで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("I love it", return_tensors="pt")

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

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

📄 ライセンス

このモデルはApache-2.0ライセンスの下で提供されています。

このモデルカードは、Manuel Romero/@mrm8488による "mrm8488/t5-base-finetuned-imdb-sentiment" に基づいています。