我正在参与「兔了个兔」构思投稿大赛，概况请看：「兔了个兔」构思投稿大赛

前言

这不兔年到了，薅薅兔毛；

昨日帮小兔子完成课后作业，作了一首包含 “兔” 字的古诗，小兔子也因而受到了老师的表彰，具体能够阅览这篇文章：【AI】用 GPT 帮小兔子写首诗；

今天，小兔子的老师又给她布置了新的课后作业，新春降临之际，写一篇文章来记载一下自己的新春生活；

所以懒散的小兔子又第一时间找到了博主，让博主用 AI 帮她写一写；

剧情需求，作业仍是要自己写的！

正文

言归正传，这里用的是一个12层的小模型，因而精度也相对较低；

仍是先加载相应的 tokenizer 以及预训练模型：

from transformers import GPT2Tokenizer,GPT2LMHeadModel
check_point = ''
tokenizer = GPT2Tokenizer.from_pretrained(check_point)
model = GPT2LMHeadModel.from_pretrained(check_point)

输入一段文本，因为小兔子的作文是要记载一下自己的新春生活，因而，咱们能够给个开头 “大年初一，”，如下：

text = "大年初一，"

然后便是将输入文本交给 tokenizer 进行预处理，传入模型 model，对于模型返回的成果进行处理，得到最终成果：

inputs = tokenizer(text,return_tensors='pt')
outputs = model.generate(**inputs, ...)
for idx, sentence in enumerate(outputs.sequences):
    print(f"{idx+1}、", tokenizer.decode(sentence).split('<|endoftext|>')[0], '\n')

【AI】用 GPT 帮小兔子写文章

能够看到成果其实不是那么的精准，有些看着就奇奇怪怪的，究竟模型比较小，参数少，不过能够多输出一些成果，总有比较合适的；

后记

以上便是【AI】用 GPT 帮小兔子写文章的全部内容了。

期望本篇博文对我们有所帮助！

系列文章：

  • 【AI】用 GPT 帮小兔子作首诗
  • 【AI】用 GPT 帮小兔子写文章
  • 【AI】用 GPT 帮小兔子画肖像

上篇精讲：【AI】用 GPT 帮小兔子写首诗

我是，等待你的重视；

创作不易，请多多支撑；

系列专栏：AI NLP