摘要
本说明书公开了一种模型训练方法、装置、存储介质及电子设备。获取样本风控任务对应的样本图数据,以及样本图数据对应的样本标注图,将所述样本图数据输入到待训练模型中,通过所述待训练模型的提示子网,确定与所述样本图数据对应的样本提示图,以使所述待训练模型的任务子网在所述样本提示图的提示下,根据所述样本图数据输出预测图数据,根据所述预测图数据和所述样本标注图的差异,对所述提示子网对应的提示参数和所述任务子网对应的任务参数进行调整,得到用于处理所述样本风控任务的风控模型。通过这种方法,能够使待训练模型在样本提示图的提示下,基于少量的样本图数据以及样本标注图,训练得到用于处理样本风控任务的风控模型。
技术关键词
预训练模型
样本
数据
模型训练方法
参数
节点
模型训练装置
电子设备
神经网络模型
处理器
输入模块
可读存储介质
存储器
关系
计算机
程序