如何load自己定制的loss函数模型
2019-03-13 本文已影响0人
夕宝爸爸
处理已保存模型中的自定义层或其他自定义对象
自己在使用magpie的时候定义了一个loss函数，为了解决类别不平衡的问题
from magpie import Magpie
from magpie.nn.models import weighted_loss
magpie = Magpie()
magpie.load_model('./dj.h5', custom_objects={'weighted_loss': weighted_loss})
