0 介绍
paddlepaddle是百度开源的一套机器学习(AI)框架
Senta_lstm是基于lstm模型、采用paddlepaddle框架实现的nlp模型,支持语义分析和情感分析
SKEP是上述模型的改进版
paddlehub集成了模型的预训练结果,提供开箱可用的模型。
1 安装
sudo pip3 install paddlepaddle -i https://mirror.baidu.com/pypi/simple
sudo pip3 install paddlehub -i https://mirror.baidu.com/pypi/simple
2 情感分析(lstm)
import paddlehub as hub
# load model
senta = hub.Module(name="senta_lstm")
# text
test_text = [
"不错呦",
"哎,一般"
]
# classify
results = senta.sentiment_classify(data={"text": test_text})
# result
for result in results:
print(result)
#
{'text': '不错呦', 'sentiment_label': 1, 'sentiment_key': 'positive', 'positive_probs': 0.9924, 'negative_pr