[发明专利]基于人工智能的图书分类装置、方法、设备及存储介质

摘要

本申请涉及预测模型领域，提供一种基于人工智能的图书分类装置、方法、设备及存储介质，该装置包括输入模块、问题模块、片段记忆模块和回答模块；输入模块用于从获取待分类的图书中获取图书的特征信息；将特征信息转换为词向量，向片断记忆模块输入词向量，向问题模块输入图书分类关键词的问题向量；问题模块用于从输入模块接收问题向量，向片断记忆模块输入所述问题向量；片段记忆模块用于采用关注力机制，根据问题向量、片断记忆模块中的历史记忆和历史问题，在输入模块输出的词向量上进行迭代，产生新记忆，得到记忆特征训练模型；回答模块用于根据记忆特征训练模型预测图书类别，并据新记忆生成与问题向量对应的回答。通过采用本方案，能够提高图书类别划分的准确率。

主权项

1.一种图书分类装置，其特征在于，所述装置包括输入模块、问题模块、片段记忆模块和回答模块；所述输入模块，用于从获取待分类的图书中获取图书的特征信息；将特征信息转换为词向量，向所述片断记忆模块输入所述词向量，以及向所述问题模块输入图书分类关键词的问题向量；所述问题模块，用于从所述输入模块接收所述问题向量，向所述片断记忆模块输入所述问题向量；所述片段记忆模块，用于采用关注力机制，根据所述问题向量、所述片断记忆模块中的历史记忆和历史问题，在所述输入模块输出的所述词向量上进行迭代，迭代产生新记忆，得到记忆特征训练模型；所述回答模块，用于根据所述记忆特征训练模型预测图书类别，并根据所述新记忆生成与所述问题向量对应的回答。