update ltp demo with srl. xuming 20180223

xuming06 · xuming06 · commit b8ff0b19b3ef · 2018-02-23T10:47:56.000+08:00
diff --git a/tool/ltp_demo.py b/tool/ltp_demo.py
@@ -4,6 +4,8 @@
 import os
 
 ltp_data_dir = '/Users/xuming06/Codes/ltp_data_v3.4.0'
+
+# segment
 cws_model_path = os.path.join(ltp_data_dir, 'cws.model')
 
 from pyltp import Segmentor
@@ -14,12 +16,14 @@
 words = segmentor.segment(text)
 print(" ".join(words))  # 我 是 中国 人 ， 我 在 爱斯基摩 打雪仗 。 欧几 里 得 是 西元前 三 世纪 的 希腊 数学家
 
+# segment with lexicon
 segmentor = Segmentor()
 # load self dictionary
 segmentor.load_with_lexicon(cws_model_path, './self_dict.txt')
 words = segmentor.segment(text)
 print(" ".join(words))  # 我 是 中国 人 ， 我 在 爱斯基摩 打雪仗 。 欧几里得 是 西元前 三 世纪 的 希腊 数学家
 
+# pos
 pos_model_path = os.path.join(ltp_data_dir, 'pos.model')
 
 from pyltp import Postagger
@@ -34,6 +38,8 @@
 word_pos = list(i[0] + '/' + i[1] for i in zipped)
 print(' '.join(word_pos))
 
+
+# ner
 ner_model_path = os.path.join(ltp_data_dir, 'ner.model')
 from pyltp import NamedEntityRecognizer
 
@@ -45,6 +51,8 @@
 
 print(' '.join(nertags))
 
+
+# parser
 par_model_path = os.path.join(ltp_data_dir, 'parser.model')
 from pyltp import Parser
 
@@ -61,6 +69,17 @@
 for i in range(len(words)):
     print(relation[i] + '(' + words[i] + ', ' + heads[i] + ')')
 
+# SRL
+srl_model_path = os.path.join(ltp_data_dir,'pisrl.model')
+from pyltp import SementicRoleLabeller
+srl = SementicRoleLabeller()
+srl.load(srl_model_path)
+roles = srl.label(words,postags,arcs)
+for role in roles:
+    print(role.index, "".join(
+        ["%s:(%d,%d)" % (arg.name, arg.range.start,arg.range.end) for arg in role.arguments]))
+
+srl.release()
 
 # jieba
 import jieba
