摘要

本文提出了一种新型的分子毒性预测模型,采用堆叠式自编码器(SAE)对高维的稀疏数据进行特征提取,而后对分类器进行改进,使用胶囊网络作为分类器进行毒性预测分类,为了证明该模型的有效性,我们将SAE+CAPS与传统的softmax分类器和SAE+softmax分类器进行了对比,实验结果表明相较于另外两种模型而言该模型拥有更高的性能。