senta_demo.py 680 字节
Newer Older
S
Steffy-zxf 已提交
1 2 3 4
#coding:utf-8
from __future__ import print_function

import json
Z
Zeyu Chen 已提交
5
import os
S
Steffy-zxf 已提交
6 7
import six

Z
Zeyu Chen 已提交
8 9 10 11
import paddlehub as hub

if __name__ == "__main__":
    # Load Senta-BiLSTM module
Z
Zeyu Chen 已提交
12
    senta = hub.Module(name="senta_bilstm")
Z
Zeyu Chen 已提交
13 14 15 16 17 18

    test_text = ["这家餐厅很好吃", "这部电影真的很差劲"]

    input_dict = {"text": test_text}

    results = senta.sentiment_classify(data=input_dict)
S
Steffy-zxf 已提交
19 20 21

    for index, text in enumerate(test_text):
        results[index]["text"] = text
Z
Zeyu Chen 已提交
22
    for index, result in enumerate(results):
S
Steffy-zxf 已提交
23
        if six.PY2:
S
shenyuhan 已提交
24 25
            print(
                json.dumps(results[index], encoding="utf8", ensure_ascii=False))
S
Steffy-zxf 已提交
26 27
        else:
            print(results[index])