对神经网络中节点进行染色的方法及装置

摘要

本发明提出一种对神经网络中节点进行染色的方法及装置，其中，方法包括：获取训练数据，并对所述训练数据进行预处理；通过预设的特征提取网络提取预处理后的训练数据对应的特征，并通过话的分类网络对所述特征进行分类，输出所述分类对应的类别向量；将所述特征提取网络和所述分类网络的节点参数按照顺序存储于预先建立的节点存储区；遍历所述节点存储区的节点信息，并对节点进行标签分类，并对各类别的节点进行染色操作。该方法通过对完成训练的深度神经网络模型进行染色，实现对神经网络中节点作用的研究，解决现有技术中深度神经网络中各个节点的可解释性问题。