![如何在Keras中使用fit_generator()加权类?[关闭],第1张 如何在Keras中使用fit_generator()加权类?[关闭],第1张](/aiimages/%E5%A6%82%E4%BD%95%E5%9C%A8Keras%E4%B8%AD%E4%BD%BF%E7%94%A8fit_generator%EF%BC%88%EF%BC%89%E5%8A%A0%E6%9D%83%E7%B1%BB%EF%BC%9F%5B%E5%85%B3%E9%97%AD%5D.png)
根据文档,我们
class_weight在
fit_generator(
Keras v.2.2.2 )中:
Class_weight
:可选的字典,将类索引(整数)映射到权重(浮点)值,用于对损失函数加权(仅在训练过程中)。这可能有助于告诉模型“更多关注”来自代表性不足的类的样本。
假设您有两个类[ 正 和 负 ],则可以传递
class_weight给
fit_generatorwith:
model.fit_generator(gen,class_weight=[0.7,1.3])
欢迎分享,转载请注明来源:内存溢出
微信扫一扫
支付宝扫一扫
评论列表(0条)