Python Keras:fit_生成器中每个训练批次的等级_重量

我使用Keras解决分割问题。我使用

fit_generator

来训练我的模型。如果

安装发电机

，我不确定如何使用

分级重量

？ 实际上，我想计算每个批次的类权重，而不是整个数据集。我无法找到一种使用

fit_generator

而不重复工作的方法

我试着用一个生成器来返回每批货物的重量，但这给了我

TypeError:类型为“generator”的对象没有len（）

我还考虑使用自定义回调来处理类权重。据我所知，Keras回调需要使用培训批次和标签进行初始化。是否有一种方法可以将我的训练批传递给我的

fit\u生成器调用中的回调

谢谢。
一种解决方法是计算网络中的类权重，并将其作为输出添加到模型中。然后创建一个自定义损失，您可以在其中访问它。我可以理解如何计算模型中的类权重。你能再解释一下海关损失吗？