gpt2-horoscopes - 基于GPT-2的星座预测，探索五大类星座运势

gpt2-horoscopes项目介绍

gpt2-horoscopes项目是一个基于GPT-2模型的有趣应用，它专注于根据不同类别生成每日星座运势。这个项目利用从Horoscopes.com上抓取的数据，对GPT-2进行了微调，使其能够在一系列特定类别下生成相应的星座运势内容。

模型描述

gpt2-horoscopes使用一个经过微调的GPT-2模型，这个模型专门训练于特定的星座运势数据集。用户可以通过输入星座运势的类别，让模型生成对应的星座运势。支持的类别包括：通用（general）、事业（career）、爱情（love）、健康（wellness）和生日（birthday）。

使用方法

用户可以使用HuggingFace的pipeline API来直接调用此模型。只需引入相关的预训练模型和tokenizer，即可开始生成内容。下列代码展示了如何初始化模型和tokenizer：

from transformers import AutoTokenizer, AutoModelWithLMHead

tokenizer = AutoTokenizer.from_pretrained("shahp7575/gpt2-horoscopes")
model = AutoModelWithLMHead.from_pretrained("shahp7575/gpt2-horoscopes")

文本生成

要生成星座运势，需要按照指定格式输入文本，即<|category|> {category_type} <|horoscope|>。例如，要生成事业运势，可以使用如下代码:

prompt = "<|category|> career <|horoscope|>"
prompt_encoded = torch.tensor(tokenizer.encode(prompt)).unsqueeze(0)
sample_outputs = model.generate(prompt_encoded, 
                                do_sample=True,   
                                top_k=40, 
                                max_length=300,
                                top_p=0.95,
                                temperature=0.95,
                                num_return_sequences=1)

为了参考，可以使用generation script。