已mac为例,windows、linux 移步官方
python版本必须在,3.6.x - 3.9.x
pip版本必须 >= 20.2.2
python -m pip install paddlepaddle==0.0.0 -f https://www.paddlepaddle.org.cn/whl/mac/cpu/develop.html
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)
雨打花残地痛心,
雨打花残地皱眉,
雨打花残地动容,
雨打霜欺地动容,
雨打花残地洒愁.