Gpt2 Emotion

🚀 GPT - 2情感文本生成模型

本項目基於GPT - 2模型，可根據指定的情感標籤生成相關的英文文本，為情感文本生成任務提供了便捷的解決方案。

🚀 快速開始

你可以使用以下代碼快速體驗模型的文本生成功能：

from transformers import pipeline

generator = pipeline('text-generation', 'heegyu/gpt2-emotion')

prompt = "sadness I'm so " # start token should be one of ["sadness", "joy", "love", "anger", "fear", "surprise"]
print(generator(prompt)[0]['generated_text'])
>>> sadness I'm so tired of seeing all the stupid things that i ve learned from past years that i feel like ive been so stupid and blah and then i feel like ive just wasted all my energy doing stupid shit like how i never

✨ 主要特性

• 情感驅動：支持基於不同情感標籤（悲傷、喜悅、愛、憤怒、恐懼、驚訝）生成文本。
• 易於使用：藉助transformers庫的pipeline函數，簡單幾行代碼即可實現文本生成。

📦 安裝指南

本項目依賴於transformers庫，你可以使用以下命令進行安裝：

pip install transformers

💻 使用示例

基礎用法

from transformers import pipeline

# 加載文本生成管道
generator = pipeline('text-generation', 'heegyu/gpt2-emotion')

# 定義輸入提示
prompt = "sadness I'm so "
# 生成文本
print(generator(prompt)[0]['generated_text'])

📚 詳細文檔

輸入提示要求

輸入提示的起始詞必須是以下之一：["sadness", "joy", "love", "anger", "fear", "surprise"]，以指定生成文本的情感傾向。

🔧 技術細節

本項目使用的模型為GPT - 2，它是一個基於Transformer架構的大型語言模型。通過在emotion數據集上進行訓練，使得模型能夠根據不同的情感標籤生成相關的文本。

📄 許可證

本項目採用MIT許可證。