一种基于Ricci曲率加权的蛋白质-配体对接方法

本申请涉及一种基于Ricci曲率加权的蛋白质‑配体对接方法。该方法包括以下步骤：获取训练数据集；选择训练数据集中的一组数据，作为第一数据；利用Ricci曲率、ESM‑2、TorchDrug方法，将所述第一数据编码转化，得到第一蛋白质配体图；将第一蛋白质配体图输入蛋白质‑配体对接模型，得到残基配体图、口袋中心与口袋半径、蛋白质‑配体对接结果；根据所述残基配体图、口袋中心与口袋半径、蛋白质‑配体对接结果、第一蛋白质配体图计算损失函数；并依据损失函数更新蛋白质‑配体对接模型；选择训练数据集中的另一组数据，作为新的第一数据，重复训练，直到损失函数降到最低；得到训练好的蛋白质‑配体对接模型；将待预测的数据进行编码转化，得到第二蛋白质配体图；将所述第二蛋白质配体图输入训练好的蛋白质‑配体对接模型，得到最终蛋白质‑配体对接结果。本发明对接的准确性高。