所用到的模型是百度的开源框架paddlehub,关于这个的先容,可以看这里的文档:
https://gitee.com/yiweilu/PaddleHub/tree/release/v1.8
可以做打算机视觉,自然措辞处理,比如提取笔墨,检测人脸是否带了口罩等等。
今日带来的项目便是根据关键词写出几句情话来,代码如下:
import paddlehub as hubmodule = 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[0])
天生的结果:
阳光,赐我微笑的力量,就像你给的力量一样。彩虹,赐我乐不雅观的心态,就像你给的心态一样。
沙滩,你可曾记得?下雨天,大概会想起我,可是你却永久不会忘却我。
贝壳,终于下雨了,你说你要离开我,由于你对付我来说,非常主要。
觉得不是特殊好,可以自己再轻微修正修正。
须要把稳的是, 运行这个脚本时,须要首先安装paddlehub
# 在命令行中输入以下命令# 确认当前所用的pip是否是paddle_env环境下的pipwhere pip# 默认安装CPU版本,安装paddle时建议利用百度源pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
安装完paddle后,连续在paddle_env环境中安装paddlehub:
# 在命令行中输入以下命令pip install paddlehub -i https://mirror.baidu.com/pypi/simple
详细安装可以参考:
https://github.com/PaddlePaddle/PaddleHub/blob/release/v2.1/docs/docs_ch/get_start/windows_quickstart.md