“基于医疗知识图谱的问答系统”代码解析（三）

“基于医疗知识图谱的问答系统”代码解析（三）

question_parser.py —问题分析代码解析

“基于医疗知识图谱的问答系统”代码解析（一）
“基于医疗知识图谱的问答系统”代码解析（二）
“基于医疗知识图谱的问答系统”代码解析（四）
“基于医疗知识图谱的问答系统”代码解析（五）

#!/usr/bin/env python3
# coding: utf-8
# File: question_parser.py
# Author: lhy<lhy_in_blcu@126.com,https://huangyong.github.io>
# Date: 18-10-4

class QuestionPaser:

    '''构建实体节点'''
    # args 实质就是将函数传入的参数，存储在元组类型的变量args当中
    def build_entitydict(self, args):
        entity_dict = {
   }
        # 键值和类型
        for arg, types in args.items():
            # 遍历类型
            for type in types:
                # 若类型不在实体字典中，则添加
                if type not in entity_dict:
                    entity_dict[type] = [arg]
                else:
                    entity_dict[type].append(arg)
        return entity_dict

    '''解析主函数'''
    def parser_main(self, res_classify):
        args = res_classify['args']
        entity_dict = self.build_entitydict(args)       # 建立实体
        question_types = res_classify['question_types'] # 建立问题类别
        sqls = []                                       # 初始化
        for question_type in question_types:
            sql_ = {
   }
            sql_['question_type'] = question_type       # 将问题类别给sql_
            sql = []
            # 查找问题类型，将问题类型里的信息给sql 需要通过cypher语句查询，自己整理返回
            if question_type == 'disease_symptom':
                sql = self.sql_transfer(question_type, entity_dict.get('disease'))

            elif question_type == 'symptom_disease':
                sql = self.sql_transfer(question_type, entity_dict.get('symptom'))

            elif question_type == 'disease_cause':
                sql = self.sql_transfer(question_type, entity_dict.get('disease')