根据文档,我们
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])

根据文档,我们
class_weight在
fit_generator(
Class_weight
:可选的字典,将类索引(整数)映射到权重(浮点)值,用于对损失函数加权(仅在训练过程中)。这可能有助于告诉模型“更多关注”来自代表性不足的类的样本。
假设您有两个类[ 正 和 负 ],则可以传递
class_weight给
fit_generatorwith:
model.fit_generator(gen,class_weight=[0.7,1.3])