如何编写长尾词的Keras自定义损失函数并加载模型？

本文共计666个文字，预计阅读时间需要3分钟。

Keras自定义函数时，通常在模型中编写好自定义函数，然后在模型编译时添加接口。例如，focal_loss和fbeta_score是我们自定义的两个函数，在model.compile时加入即可。

keras自定义函数时候，正常在模型里自己写好自定义的函数，然后在模型编译的那行代码里写上接口即可。如下所示，focal_loss和fbeta_score是我们自己定义的两个函数，在model.compile加入它们，metrics里‘accuracy'是keras自带的度量函数。

def focal_loss(): ... return xx def fbeta_score(): ... return yy model.compile(optimizer=Adam(lr=0.0001), loss=[focal_loss],metrics=['accuracy',fbeta_score] )

训练好之后，模型加载也需要再额外加一行，通过load_model里的custom_objects将我们定义的两个函数以字典的形式加入就能正常加载模型啦。