分类：模型加载

2020
10-08

keras自定义损失函数并且模型加载的写法介绍

keras自定义函数时候，正常在模型里自己写好自定义的函数，然后在模型编译的那行代码里写上接口即可。如下所示，focal_loss和fbeta_score是我们自己定义的两个函数，在model.compile加入它们，metrics里‘accuracy'是keras自带的度量函数。deffocal_loss():...returnxxdeffbeta_score():...returnyymodel.compile(optimizer=Adam(lr=0.0001),loss=[focal_loss],metrics=['accuracy',fbeta_score])训练好之后，模型加载也... 继续阅读 >