结合嵌入模型和指令微调大模型的实体预测方法及装置

本申请提出了一种结合嵌入模型和指令微调大模型的实体预测方法及装置，包括：获取实体预测的数据集，将其分割为嵌入模型训练数据、微调指令生成数据和超参数调整数据；通过嵌入模型训练数据训练嵌入模型；通过微调指令生成数据和嵌入模型，填充微调指令模板，得到微调指令数据集；使用微调指令数据集对大模型进行微调，在微调时，使用超参数调整数据调整大模型的超参数，选出最优的超参数组合和最优的微调大模型；给定实体预测任务，通过嵌入模型生成候选实体列表，并使用微调大模型从候选实体列表中选出预测实体。采用上述方案的本发明提升了知识图谱补全任务的准确性和可靠性。