本文主要是推荐几个百度飞浆开源项目

情话生成模型

藏头诗生成模型

诗歌生成模型

对联生成模型

安装环境

以mac为例，windows、linux 移步官方

一、准备环境

1.目前飞桨支持的环境

• macOS 版本 10.11/10.12/10.13/10.14 (64 bit) (不支持GPU版本)
• Python 版本 3.6/3.7/3.8/3.9 (64 bit)
• pip 或 pip3 版本 20.2.2或更高版本 (64 bit)
• PaddlePaddle 不支持 M1 芯片 （arm64架构）

2.python版本

python --version

python版本必须在，3.6.x - 3.9.x

3.pip版本

python -m pip --version

pip版本必须 >= 20.2.2

二、开始安装

1.安装

python -m pip install paddlepaddle==0.0.0 -f 

https://www.paddlepaddle.org.cn/whl/mac/cpu/develop.html

2.验证

import paddle

paddle.utils.run_check()

如果出现PaddlePaddle is installed successfully!，说明您已成功安装。

项目展示

一、情话生成模型

代码

import paddlehub as hub
module = hub.Module(name="ernie_gen_lover_words")
test_texts = ['情人节', '故乡', '我爱你你爱我']
results = module.generate(texts=test_texts, use_gpu=True, beam_width=5)
for result in results:
    print(result)

效果

二、藏头诗生成模型

代码

import paddlehub as hub


# 在模型定义时，可以通过设置line=4或8指定输出绝句或律诗，设置word=5或7指定输出五言或七言。
# 默认line=4, word=7 即输出七言绝句。
module = hub.Module(name="ernie_gen_acrostic_poetry", line=4, word=7)


test_texts = ['掘金最牛']
results = module.generate(texts=test_texts, use_gpu=True, beam_width=5)
for result in results:
    print(result)

效果

掘沟引水浇蔬圃，
金谷无根种药苗。
最好移根来旧室，
牛头上午午鸡啼。
复制代码
掘沟引水浇蔬圃，
金谷无根种药苗。
最苦一年春雨歇，
牛鸠无数满仓埋。

三、诗歌生成模型

代码

import paddlehub as hub


module = hub.Module(name="ernie_gen_poetry")


test_texts = ['床前明月光，李白是卷王。']
results = module.generate(texts=test_texts, use_gpu=True, beam_width=5)
for result in results:
    print(result)

效果

卷王书在床，月在床上床。
床上床上琴，琴声弹一曲。
琴声弹一曲，月色明一曲。
曲终各分散，露滴芭蕉绿。
复制代码
卷王书在床，月在床上床。
床上床上琴，琴声弹一曲。
琴声弹一曲，月色明一曲。
曲罢月欲落，桂树风飕飕。
复制代码
卷王书在床，月在床上床。
床上床上琴，琴声弹一曲。
琴声弹一曲，月色明一曲。
曲罢月欲落，桂树枝上宿。

四、对联生成模型

代码

import paddlehub as hub


module = hub.Module(name="ernie_gen_couplet")


test_texts = ["风吹云乱天垂泪"]
results = module.generate(texts=test_texts, use_gpu=True, beam_width=5)
for result in results:
    print(result)

效果

雨打花残地痛心,

雨打花残地皱眉,

雨打花残地动容,

雨打霜欺地动容,

雨打花残地洒愁.

作者：
链接：https://juejin.cn/post/6989026845005971493
来源：掘金
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。