一种基于大语言模型嵌入层的药物不良反应预测系统

本发明涉及一种基于大语言模型嵌入层的药物不良反应预测系统，包括：训练样本构建模块和预测模块。其中，训练样本模块包括第一数据输入单元和第一数据处理单元；预测模块包括第一数据读取单元、T‑TA多头注意力机制单元、解码器和输出单元；T‑TA多头注意力机制单元中的嵌入层，为医疗Llama模型使用药物分子数据集训练后得到的第一嵌入层。第二数据输入单元采集药物名称和对应的分子属性数据，包括分子结构、分子大小、手性和毒性数据，第二数据处理单元进行清洗、去重和结构化操作，得到药物分子数据集。与现有技术相比，本发明具有预测能力强、实时性高和准确性高等优点。