湖南响应式网站建设推荐,国外那些视频网站做的不错,房价在2022年大跌,爱奇艺会员推广联盟前言 Neo4j的数据库构建完成后#xff0c;现在就是要实现医疗知识的解答功能了。因为是初版#xff0c;这里的问题解答不会涉及深度学习#xff0c;目前只是一个条件查询的过程。而这个过程包括对问题的关键词拆解分类#xff0c;然后提取词语和类型去图数据库查询#xf…前言 Neo4j的数据库构建完成后现在就是要实现医疗知识的解答功能了。因为是初版这里的问题解答不会涉及深度学习目前只是一个条件查询的过程。而这个过程包括对问题的关键词拆解分类然后提取词语和类型去图数据库查询最后就是根据查询结果和问题类型组装语言完成回答那么以下就是完成这个过程的全部代码流程了。 环境 这里所需的环境除了前面提到的外还需要ahocorasick库用于从问题中提取关键词。另一个是colorama用于给输出面板文字美化的库。 编码
1. 问答面板
from colorama import init,Fore,Style,Back
from classifier import Classifier
from parse import Parse
from answer import Answerclass ChatRobot:def __init__(self):init(autoresetTrue)print()print(Back.BLUE欢迎进入智慧医疗问答面板)print()def main(self, question):print()default_answer 您好小北知识有限暂时回答不上来正在努力迭代中final_classify Classifier().classify(question)parse_sql Parse().main(final_classify)final_answer Answer().main(parse_sql)if not final_answer:return default_answerreturn \n\n.join(final_answer)if __name__ __main__:robot ChatRobot()while 1:print( )question input(您问)if 关闭 in question:print()print(小北说, 好的已经关闭了哦欢迎您下次提问~)break;answer robot.main(question)print(Fore.LIGHTRED_EX小北答, Fore.GREEN answer)
2. 问题归类
import ahocorasickclass Classifier:def __init__(self):# print(开始初始化, time.strftime(%Y-%m-%d %H:%M:%S, time.localtime()))self.checks_wds [i.strip() for i in open(dict/checks.txt, encodingutf-8, moder) if i.strip()]self.departments_wds [i.strip() for i in open(dict/departments.txt, encodingutf-8, moder) if i.strip()]self.diseases_wds [i.strip() for i in open(dict/diseases.txt, encodingutf-8, moder) if i.strip()]self.drugs_wds [i.strip() for i in open(dict/drugs.txt, encodingutf-8, moder) if i.strip()]self.foods_wds [i.strip() for i in open(dict/foods.txt, encodingutf-8, moder) if i.strip()]self.producers_wds [i.strip() for i in open(dict/producers.txt, encodingutf-8, moder) if i.strip()]self.symptoms_wds [i.strip() for i in open(dict/symptoms.txt, encodingutf-8, moder) if i.strip()]self.features_wds set(self.checks_wdsself.departments_wdsself.diseases_wdsself.drugs_wdsself.foods_wdsself.producers_wdsself.symptoms_wds)self.deny_words [name.strip() for name in open(dict/deny.txt, encodingutf-8, moder) if name.strip()]# actree 从输入文本中提取出指定分词表中的词self.actree self.build_actree(list(self.features_wds))# 给每个词创建类型词典相当慢的操作self.wds_dict self.build_words_dict()# print(给每个词创建类型词典结束, time.strftime(%Y-%m-%d %H:%M:%S, time.localtime()))# 问句疑问词self.symptom_qwds [症状, 表征, 现象, 症候, 表现]self.cause_qwds [原因, 成因, 为什么, 怎么会, 怎样才, 咋样才, 怎样会, 如何会, 为啥, 为何, 如何才会, 怎么才会, 会导致,会造成]self.acompany_qwds [并发症, 并发, 一起发生, 一并发生, 一起出现, 一并出现, 一同发生, 一同出现, 伴随发生, 伴随, 共现]self.food_qwds [饮食, 饮用, 吃, 食, 伙食, 膳食, 喝, 菜, 忌口, 补品, 保健品, 食谱, 菜谱, 食用, 食物, 补品]self.drug_qwds [药, 药品, 用药, 胶囊, 口服液, 炎片]self.prevent_qwds [预防, 防范, 抵制, 抵御, 防止, 躲避, 逃避, 避开, 免得, 逃开, 避开, 避掉, 躲开, 躲掉, 绕开,怎样才能不, 怎么才能不, 咋样才能不, 咋才能不, 如何才能不,怎样才不, 怎么才不, 咋样才不, 咋才不, 如何才不,怎样才可以不, 怎么才可以不, 咋样才可以不, 咋才可以不, 如何可以不,怎样才可不, 怎么才可不, 咋样才可不, 咋才可不, 如何可不]self.lasttime_qwds [周期, 多久, 多长时间, 多少时间, 几天, 几年, 多少天, 多少小时, 几个小时, 多少年]self.cureway_qwds [怎么治疗, 如何医治, 怎么医治, 怎么治, 怎么医, 如何治, 医治方式, 疗法, 咋治, 怎么办, 咋办, 咋治]self.cureprob_qwds [多大概率能治好, 多大几率能治好, 治好希望大么, 几率, 几成, 比例, 可能性, 能治, 可治, 可以治, 可以医]self.easyget_qwds [易感人群, 容易感染, 易发人群, 什么人, 哪些人, 感染, 染上, 得上]self.check_qwds [检查, 检查项目, 查出, 检查, 测出, 试出]self.belong_qwds [属于什么科, 属于, 什么科, 科室]self.cure_qwds [治疗什么, 治啥, 治疗啥, 医治啥, 治愈啥, 主治啥, 主治什么, 有什么用, 有何用, 用处, 用途,有什么好处, 有什么益处, 有何益处, 用来, 用来做啥, 用来作甚, 需要, 要]构造actree加速过滤def build_actree(self, wordlist):actree ahocorasick.Automaton()for index, word in enumerate(wordlist):actree.add_word(word, (index, word))actree.make_automaton()return actree# 构建特征词属性def build_words_dict(self):words_dict {}check_words set(self.checks_wds)department_words set(self.departments_wds)disease_words set(self.diseases_wds)drug_words set(self.drugs_wds)food_words set(self.foods_wds)producer_words set(self.producers_wds)symptom_words set(self.symptoms_wds)for word in self.features_wds:words_dict[word] []if word in check_words:words_dict[word].append(check)if word in department_words:words_dict[word].append(department)if word in disease_words:words_dict[word].append(disease)if word in drug_words:words_dict[word].append(drug)if word in food_words:words_dict[word].append(food)if word in producer_words:words_dict[word].append(producer)if word in symptom_words:words_dict[word].append(symptom)return words_dict# 根据输入返回问题类型def classify(self, sent):# 最终输入给解析器的字典data {}region_words []lists self.actree.iter(sent)for ii in lists:cur_word ii[1][1]region_words.append(cur_word)# {职业黑变病: [diseases], 倒睫: [diseases, symptom]}final_dict {i_name: self.wds_dict.get(i_name) for i_name in region_words}data[args] final_dictquestion_type otherquestions_type []# [diseases, diseases, symptom]type []for i_type in final_dict.values():type i_type# 判断type中是否有指定类型, 提出的问题是否包含指定的修饰词给问题定类型# 1. 如提问词是否出现状态词语那就是问某种疾病会出现什么症状if self.check_word_exist(self.symptom_qwds, sent) and (disease in type):question_type disease_symptomquestions_type.append(question_type)# 根据症状问疾病if self.check_word_exist(self.symptom_qwds, sent) and (symptom in type):question_type symptom_diseasequestions_type.append(question_type)# 原因if self.check_word_exist(self.cause_qwds, sent) and (disease in type):question_type disease_causequestions_type.append(question_type)# 并发症if self.check_word_exist(self.acompany_qwds, sent) and (disease in type):question_type disease_acompanyquestions_type.append(question_type)# 推荐食品if self.check_word_exist(self.food_qwds, sent) and disease in type:deny_status self.check_word_exist(self.deny_words, sent)if deny_status:question_type disease_not_foodelse:question_type disease_do_foodquestions_type.append(question_type)# 已知食物找疾病if self.check_word_exist(self.food_qwds self.cure_qwds, sent) and food in type:deny_status self.check_word_exist(self.deny_words, sent)if deny_status:question_type food_not_diseaseelse:question_type food_do_diseasequestions_type.append(question_type)# 推荐药品if self.check_word_exist(self.drug_qwds, sent) and disease in type:question_type disease_drugquestions_type.append(question_type)# 药品治啥病if self.check_word_exist(self.cure_qwds, sent) and drug in type:question_type drug_diseasequestions_type.append(question_type)# 疾病接受检查项目if self.check_word_exist(self.check_qwds, sent) and disease in type:question_type disease_checkquestions_type.append(question_type)# 已知检查项目查相应疾病if self.check_word_exist(self.check_qwds self.cure_qwds, sent) and check in type:question_type check_diseasequestions_type.append(question_type)# 症状防御if self.check_word_exist(self.prevent_qwds, sent) and disease in type:question_type disease_preventquestions_type.append(question_type)# 疾病医疗周期if self.check_word_exist(self.lasttime_qwds, sent) and disease in type:question_type disease_lasttimequestions_type.append(question_type)# 疾病治疗方式if self.check_word_exist(self.cureway_qwds, sent) and disease in type:question_type disease_curewayquestions_type.append(question_type)# 疾病治愈可能性if self.check_word_exist(self.cureprob_qwds, sent) and disease in type:question_type disease_cureprobquestions_type.append(question_type)# 疾病易感染人群if self.check_word_exist(self.easyget_qwds, sent) and disease in type:question_type disease_easygetquestions_type.append(question_type)# 若没有查到相关的外部查询信息那么则将该疾病的描述信息返回if questions_type [] and disease in type:questions_type [disease_desc]# 若没有查到相关的外部查询信息那么则将该疾病的描述信息返回if questions_type [] and symptom in type:questions_type [symptom_disease]# 将多个分类结果进行合并处理组装成一个字典data[question_types] questions_typereturn datadef check_word_exist(self, word_list, words):for item in word_list:if item in words:return Truereturn False3. 类型解析(查询组装) class Parse:def main(self, classify):entity classify[args]questions_type classify[question_types]entity_dict self.entity_transform(entity)sqls []for question in questions_type:sql_dict {}sql_dict[qustion_type] questionsql_dict[sql] []sql []if question disease_symptom:sql self.sql_transfer(question, entity_dict.get(disease))elif question symptom_disease:sql self.sql_transfer(question, entity_dict.get(symptom))elif question disease_cause:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_acompany:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_not_food:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_do_food:sql self.sql_transfer(question, entity_dict.get(disease))elif question food_not_disease:sql self.sql_transfer(question, entity_dict.get(food))elif question food_do_disease:sql self.sql_transfer(question, entity_dict.get(food))elif question disease_drug:sql self.sql_transfer(question, entity_dict.get(disease))elif question drug_disease:sql self.sql_transfer(question, entity_dict.get(drug))elif question disease_check:sql self.sql_transfer(question, entity_dict.get(disease))elif question check_disease:sql self.sql_transfer(question, entity_dict.get(check))elif question disease_prevent:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_lasttime:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_cureway:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_cureprob:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_easyget:sql self.sql_transfer(question, entity_dict.get(disease))elif question disease_desc:sql self.sql_transfer(question, entity_dict.get(disease))if sql:sql_dict[sql] sqlsqls.append(sql_dict)return sqlsdef sql_transfer(self, question_type, entities):# 查询语句sql []# 查询疾病的原因if question_type disease_cause:sql [MATCH (m:Diseases) where m.name {0} return m.name, m.cause.format(i) for i in entities]# 查询疾病的防御措施elif question_type disease_prevent:sql [MATCH (m:Diseases) where m.name {0} return m.name, m.prevent.format(i) for i in entities]# 查询疾病的持续时间elif question_type disease_lasttime:sql [MATCH (m:Diseases) where m.name {0} return m.name, m.cure_lasttime.format(i) for i in entities]# 查询疾病的治愈概率elif question_type disease_cureprob:sql [MATCH (m:Diseases) where m.name {0} return m.name, m.cured_prob.format(i) for i in entities]# 查询疾病的治疗方式elif question_type disease_cureway:sql [MATCH (m:Diseases) where m.name {0} return m.name, m.cure_way.format(i) for i in entities]# 查询疾病的易发人群elif question_type disease_easyget:sql [MATCH (m:Diseases) where m.name {0} return m.name, m.easy_get.format(i) for i in entities]# 查询疾病的相关介绍elif question_type disease_desc:sql [MATCH (m:Diseases) where m.name {0} return m.name, m.desc.format(i) for i in entities]# 查询疾病有哪些症状elif question_type disease_symptom:sql [MATCH (m:Diseases)-[r:has_symptoms]-(n:Symptoms) where m.name {0} return m.name, r.name, n.name.format(i) for i in entities]# 查询症状会导致哪些疾病elif question_type symptom_disease:sql [MATCH (m:Diseases)-[r:has_symptoms]-(n:Symptoms) where n.name {0} return m.name, r.name, n.name.format(i) for i in entities]# 查询疾病的并发症elif question_type disease_acompany:sql1 [MATCH (m:Diseases)-[r:acompany_with]-(n:Symptoms) where m.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql2 [MATCH (m:Diseases)-[r:acompany_with]-(n:Symptoms) where n.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql sql1 sql2# 查询疾病的忌口elif question_type disease_not_food:sql [MATCH (m:Diseases)-[r:not_eat]-(n:Foods) where m.name {0} return m.name, r.name, n.name.format(i)for i in entities]# 查询疾病建议吃的东西elif question_type disease_do_food:sql1 [MATCH (m:Diseases)-[r:do_eat]-(n:Foods) where m.name {0} return m.name, r.name, n.name.format(i)for i in entities]sql2 [MATCH (m:Diseases)-[r:recomment_eat]-(n:Foods) where m.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql sql1 sql2# 已知忌口查疾病elif question_type food_not_disease:sql [MATCH (m:Diseases)-[r:not_eat]-(n:Foods) where n.name {0} return m.name, r.name, n.name.format(i)for i in entities]# 已知推荐查疾病elif question_type food_do_disease:sql1 [MATCH (m:Diseases)-[r:do_eat]-(n:Foods) where n.name {0} return m.name, r.name, n.name.format(i)for i in entities]sql2 [MATCH (m:Diseases)-[r:recomment_eat]-(n:Foods) where n.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql sql1 sql2# 查询疾病常用药品药品别名记得扩充elif question_type disease_drug:sql1 [MATCH (m:Diseases)-[r:common_drug]-(n:Drugs) where m.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql2 [MATCH (m:Diseases)-[r:recommand_drug]-(n:Drugs) where m.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql sql1 sql2# 已知药品查询能够治疗的疾病elif question_type drug_disease:sql1 [MATCH (m:Diseases)-[r:common_drug]-(n:Drugs) where n.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql2 [MATCH (m:Diseases)-[r:recommand_drug]-(n:Drugs) where n.name {0} return m.name, r.name, n.name.format(i) for i in entities]sql sql1 sql2# 查询疾病应该进行的检查elif question_type disease_check:sql [MATCH (m:Diseases)-[r:need_check]-(n:Checks) where m.name {0} return m.name, r.name, n.name.format(i) for i in entities]# 已知检查查询疾病elif question_type check_disease:sql [MATCH (m:Diseases)-[r:need_check]-(n:Checks) where n.name {0} return m.name, r.name, n.name.format(i) for i in entities]return sqldef entity_transform(self, entity):entity_dict {}for args, types in entity.items():for type in types:if type in entity_dict:entity_dict[type] [args]else:entity_dict[type] []entity_dict[type].append(args)return entity_dict
4. 数据查询(回答组装)
from py2neo import Graph, Nodeclass Answer:def __init__(self):self.neo4j Graph(bolt://localhost:7687, auth(neo4j, beiqiaosu123456))self.num_limit 20def main(self, question_parse):answers_final []for item in question_parse:question_type item[qustion_type]sqls item[sql]answer []for sql in sqls:data self.neo4j.run(sql)answerdata.data()final_answer self.answer_prettify(question_type, answer)if final_answer:answers_final.append(final_answer)return answers_final根据对应的qustion_type调用相应的回复模板def answer_prettify(self, question_type, answers):final_answer []if not answers:return if question_type disease_symptom:desc [i[n.name] for i in answers]subject answers[0][m.name]final_answer {0}的症状包括{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type symptom_disease:desc [i[m.name] for i in answers]subject answers[0][n.name]final_answer 症状{0}可能染上的疾病有{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_cause:desc [i[m.cause] for i in answers]subject answers[0][m.name]final_answer {0}可能的成因有{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_prevent:desc [i[m.prevent] for i in answers]subject answers[0][m.name]final_answer {0}的预防措施包括{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_lasttime:desc [i[m.cure_lasttime] for i in answers]subject answers[0][m.name]final_answer {0}治疗可能持续的周期为{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_cureway:desc [;.join(i[m.cure_way]) for i in answers]subject answers[0][m.name]final_answer {0}可以尝试如下治疗{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_cureprob:desc [i[m.cured_prob] for i in answers]subject answers[0][m.name]final_answer {0}治愈的概率为仅供参考{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_easyget:desc [i[m.easy_get] for i in answers]subject answers[0][m.name]final_answer {0}的易感人群包括{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_desc:desc [i[m.desc] for i in answers]subject answers[0][m.name]final_answer {0},熟悉一下{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_acompany:desc1 [i[n.name] for i in answers]desc2 [i[m.name] for i in answers]subject answers[0][m.name]desc [i for i in desc1 desc2 if i ! subject]final_answer {0}的症状包括{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_not_food:desc [i[n.name] for i in answers]subject answers[0][m.name]final_answer {0}忌食的食物包括有{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_do_food:do_desc [i[n.name] for i in answers if i[r.name] 可以吃]recommand_desc [i[n.name] for i in answers if i[r.name] 推荐吃]subject answers[0][m.name]final_answer {0}宜食的食物包括有{1}\n推荐食谱包括有{2}.format(subject, ;.join(list(set(do_desc))[:self.num_limit]),;.join(list(set(recommand_desc))[:self.num_limit]))elif question_type food_not_disease:desc [i[m.name] for i in answers]subject answers[0][n.name]final_answer 患有{0}的人最好不要吃{1}.format(.join(list(set(desc))[:self.num_limit]), subject)elif question_type food_do_disease:desc [i[m.name] for i in answers]subject answers[0][n.name]final_answer 患有{0}的人建议多试试{1}.format(.join(list(set(desc))[:self.num_limit]), subject)elif question_type disease_drug:desc [i[n.name] for i in answers]subject answers[0][m.name]final_answer {0}通常的使用的药品包括{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type drug_disease:desc [i[m.name] for i in answers]subject answers[0][n.name]final_answer {0}主治的疾病有{1},可以试试.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type disease_check:desc [i[n.name] for i in answers]subject answers[0][m.name]final_answer {0}通常可以通过以下方式检查出来{1}.format(subject, .join(list(set(desc))[:self.num_limit]))elif question_type check_disease:desc [i[m.name] for i in answers]subject answers[0][n.name]final_answer 通常可以通过{0}检查出来的疾病有{1}.format(subject, .join(list(set(desc))[:self.num_limit]))return final_answer 写在最后 以上就是这个医疗知识问答机器人的全部代码了从上面的问答里也能看出回答得还是很生硬。因为这就只是一个程序化得思维导图所以修改完善空间还是很大这个就要后期用深度学习得方式对分类解析部分进行改动。
