首页 > 解决方案 > 按组评估自定义损失?(凯拉斯)

问题描述

我正在尝试自定义损失函数。我之前问过一个关于向它们添加参数的问题。

这很好用。每个示例也有一个组 ID。它是 5 或 6 组,但主要是 6 个示例。

我希望它们在损失函数中一起被评估。

所以我的想法是按组 id 排序,然后将 batch_size 设为 6 的乘数,以便将组保持在一起。我知道这种方法无法照顾 5 人的小组,但我现在可以牺牲这一点。但是,如果您有更好的想法,好吗?

下一步将在 y_true 中传递组标识符并将其拆分到损失函数中。比我会在内部进行分组并在组级别上进行评估。我想对组中的预测进行排序并评估最佳示例。

我不确定是否有一百个实施,但这是我现在正在做的事情。如果有更简单的方法,请分享。

标签: pythontensorflowkerasgroup-byloss-function

解决方案


推荐阅读