首页 > 解决方案 > Keras 在自定义损失函数中使用 class_weights 吗?

问题描述

简单的问题,找不到任何地方:

自定义损失函数示例:

def player_loss(self, y_true, y_pred): 
     return K.categorical_crossentropy(y_true,y_pred)

这个损失函数会在自动调用时使用class_weights提供给fit_gen方法的 if 吗?或者如果您使用自定义损失函数,您必须编写一个手动使用权重的损失函数?

标签: tensorflowkeras

解决方案


推荐阅读