Keras:fit_generator中每个训练批次的class_weight

时间:2018-05-18 07:08:00

标签: python keras deep-learning

我正在使用Keras解决分段问题。我使用<script> function myFunction() { var copyText = document.getElementById("myInput"); copyText.select(); document.execCommand("copy"); alert("Copied the text: " + copyText.value); } </script> 来训练我的模型。我不确定在fit_generator的情况下如何使用class_weight? 我实际上想要计算每个批次的类权重,而不是整个数据集。而且我无法使用fit_generator找到一种方法来做到这一点,而无需重复工作。

我尝试使用生成器为每个批次返回class_weight,但这给了我

  

TypeError:'generator'类型的对象没有len()

我还考虑过使用自定义回调来处理类权重。据我所知,Keras回调需要使用培训批次和标签进行初始化。有没有办法可以将我的训练批次从我的fit_generator电话中传递回来?

感谢。

0 个答案:

没有答案