-
Notifications
You must be signed in to change notification settings - Fork 382
/
config.py
executable file
·154 lines (150 loc) · 6.9 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
# -*- coding:utf-8 -*-
semantic_slot = {
"定义":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病) WHERE p.name='{Disease}' RETURN p.desc",
"reply_template" : "'{Disease}' 是这样的:\n",
"ask_template" : "您问的是 '{Disease}' 的定义吗?",
"intent_strategy" : "",
"deny_response":"很抱歉没有理解你的意思呢~"
},
"病因":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病) WHERE p.name='{Disease}' RETURN p.cause",
"reply_template" : "'{Disease}' 疾病的原因是:\n",
"ask_template" : "您问的是疾病 '{Disease}' 的原因吗?",
"intent_strategy" : "",
"deny_response":"您说的我有点不明白,您可以换个问法问我哦~"
},
"预防":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病) WHERE p.name='{Disease}' RETURN p.prevent",
"reply_template" : "关于 '{Disease}' 疾病您可以这样预防:\n",
"ask_template" : "请问您问的是疾病 '{Disease}' 的预防措施吗?",
"intent_strategy" : "",
"deny_response":"额~似乎有点不理解你说的是啥呢~"
},
"临床表现(病症表现)":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病)-[r:has_symptom]->(q:症状) WHERE p.name='{Disease}' RETURN q.name",
"reply_template" : "'{Disease}' 疾病的病症表现一般是这样的:\n",
"ask_template" : "您问的是疾病 '{Disease}' 的症状表现吗?",
"intent_strategy" : "",
"deny_response":"人类的语言太难了!!"
},
"相关病症":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病)-[r:acompany_with]->(q:疾病) WHERE p.name='{Disease}' RETURN q.name",
"reply_template" : "'{Disease}' 疾病的具有以下并发疾病:\n",
"ask_template" : "您问的是疾病 '{Disease}' 的并发疾病吗?",
"intent_strategy" : "",
"deny_response":"人类的语言太难了!!~"
},
"治疗方法":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : ["MATCH(p:疾病) WHERE p.name='{Disease}' RETURN p.cure_way",
"MATCH(p:疾病)-[r:recommand_drug]->(q) WHERE p.name='{Disease}' RETURN q.name",
"MATCH(p:疾病)-[r:recommand_recipes]->(q) WHERE p.name='{Disease}' RETURN q.name"],
"reply_template" : "'{Disease}' 疾病的治疗方式、可用的药物、推荐菜肴有:\n",
"ask_template" : "您问的是疾病 '{Disease}' 的治疗方法吗?",
"intent_strategy" : "",
"deny_response":"没有理解您说的意思哦~"
},
"所属科室":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病)-[r:cure_department]->(q:科室) WHERE p.name='{Disease}' RETURN q.name",
"reply_template" : "得了 '{Disease}' 可以挂这个科室哦:\n",
"ask_template" : "您想问的是疾病 '{Disease}' 要挂什么科室吗?",
"intent_strategy" : "",
"deny_response":"您说的我有点不明白,您可以换个问法问我哦~"
},
"传染性":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病) WHERE p.name='{Disease}' RETURN p.easy_get",
"reply_template" : "'{Disease}' 较为容易感染这些人群:\n",
"ask_template" : "您想问的是疾病 '{Disease}' 会感染哪些人吗?",
"intent_strategy" : "",
"deny_response":"没有理解您说的意思哦~"
},
"治愈率":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病) WHERE p.name='{Disease}' RETURN p.cured_prob",
"reply_template" : "得了'{Disease}' 的治愈率为:",
"ask_template" : "您想问 '{Disease}' 的治愈率吗?",
"intent_strategy" : "",
"deny_response":"您说的我有点不明白,您可以换个问法问我哦~"
},
"治疗时间":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病) WHERE p.name='{Disease}' RETURN p.cure_lasttime",
"reply_template" : "疾病 '{Disease}' 的治疗周期为:",
"ask_template" : "您想问 '{Disease}' 的治疗周期吗?",
"intent_strategy" : "",
"deny_response":"很抱歉没有理解你的意思呢~"
},
"化验/体检方案":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病)-[r:need_check]->(q:检查) WHERE p.name='{Disease}' RETURN q.name",
"reply_template" : "得了 '{Disease}' 需要做以下检查:\n",
"ask_template" : "您是想问 '{Disease}' 要做什么检查吗?",
"intent_strategy" : "",
"deny_response":"您说的我有点不明白,您可以换个问法问我哦~"
},
"禁忌":{
"slot_list" : ["Disease"],
"slot_values":None,
"cql_template" : "MATCH(p:疾病)-[r:not_eat]->(q:食物) WHERE p.name='{Disease}' RETURN q.name",
"reply_template" : "得了 '{Disease}' 切记不要吃这些食物哦:\n",
"ask_template" : "您是想问 '{Disease}' 不可以吃的食物是什么吗?",
"intent_strategy" : "",
"deny_response":"额~似乎有点不理解你说的是啥呢~~"
},
"unrecognized":{
"slot_values":None,
"replay_answer" : "非常抱歉,我还不知道如何回答您,我正在努力学习中~",
}
}
intent_threshold_config = {
"accept":0.8,
"deny":0.4
}
default_answer = """很抱歉我还不知道回答你这个问题\n
你可以问我一些有关疾病的\n
定义、原因、治疗方法、注意事项、挂什么科室\n
预防、禁忌等相关问题哦~"""
gossip_corpus = {
"greet":[
"hi",
"你好呀",
"我是智能医疗诊断机器人,有什么可以帮助你吗",
"hi,你好,你可以叫我小智",
"你好,你可以问我一些关于疾病诊断的问题哦"
],
"goodbye":[
"再见,很高兴为您服务",
"bye",
"再见,感谢使用我的服务",
"再见啦,祝你健康"
],
"deny":[
"很抱歉没帮到您",
"I am sorry",
"那您可以试着问我其他问题哟"
],
"isbot":[
"我是小智,你的智能健康顾问",
"你可以叫我小智哦~",
"我是医疗诊断机器人小智"
],
}