[发明专利]基于可视化的代码识别模型训练方法及装置

权利要求书

1.一种基于可视化的代码识别模型训练方法，其特征在于，包括：

显示模型配置页面；

获取用户在所述模型配置页面内输入的模型训练模式、模型网络结构和超参数；

获取与所述模型网络结构匹配的待训练代码识别模型；

根据所述模型训练模式，获取预先保存的所述待训练代码识别模型对应的代码训练样本；

加载所述超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练。

2.根据权利要求1所述的方法，其特征在于，在所述加载所述超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练之前，还包括：

获取所述用户在所述模型配置页面内输入的模型输入参数和模型输出参数；

所述加载所述超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练，包括：

根据所述模型输入参数，将所述代码训练样本输入至所述待训练代码识别模型；

根据所述模型输出参数，获取由所述待训练代码识别模型输出的所述代码训练样本对应的代码预测结果；

根据所述代码训练样本对应的初始代码结果和所述代码预测结果，计算得到所述待训练代码识别模型对应的损失值；

在所述损失值处于预设范围内的情况下，将训练后的代码识别模型作为最终的目标代码识别模型。

3.根据权利要求2所述的方法，其特征在于，在所述加载所述超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练之前，还包括：

获取所述用户在所述模型配置页面内输入的损失函数类型；

所述根据所述代码训练样本对应的初始代码结果和所述代码预测结果，计算得到所述待训练代码识别模型对应的损失值，包括：

根据所述初始代码结果和所述代码预测结果，计算得到所述待训练代码识别模型对应的与所述损失函数类型匹配的损失值。

4.根据权利要求2所述的方法，其特征在于，在所述根据所述代码训练样本对应的初始代码结果和所述代码预测结果，计算得到所述待训练代码识别模型对应的损失值之后，还包括：

在所述损失值处于所述预设范围外的情况下，获取所述损失值对应的损失梯度；

根据所述损失梯度调整所述超参数，得到目标超参数；

加载所述目标超参数，根据所述代码训练样本，迭代执行对所述待训练代码识别模型的训练过程，直至所述待训练代码识别模型对应的损失值处于所述预设范围内。

5.根据权利要求2所述的方法，其特征在于，在所述加载所述超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练之前，还包括：

校验所述模型输入参数、所述模型输出参数和所述超参数的配置有效性；

在所述配置有效性不满足模型训练条件的情况下，生成配置错误提示信息；

将所述配置错误提示信息发送给所述用户，以由所述用户根据所述配置错误提示信息对所述模型输入参数、和/或所述模型输出参数、和/或所述超参数重新进行配置。

6.根据权利要求1所述的方法，其特征在于，所述加载所述超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练，包括：

对所述超参数进行赋值，得到赋值超参数；

加载所述赋值超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练。

7.根据权利要求1所述的方法，其特征在于，在所述加载所述超参数，并根据所述代码训练样本对所述待训练代码识别模型进行训练之前，还包括：

获取所述用户在所述模型配置页面内输入的钩子函数；

所述根据所述代码训练样本对所述待训练代码识别模型进行训练，包括：

在根据所述代码训练样本对所述待训练代码识别模型的训练过程中，自动加载所述钩子函数。