【发布时间】:2021-04-26 07:32:06
【问题描述】:
我有 2 个类别的不平衡数据集。我正在使用categorical_crossentropy。我想知道我的代码。将class_weight 与categorical_crossentropy 一起使用是否正确?如果是,class_weight 是仅适用于训练集还是适用于整个数据??
我搜索了很多次,但没有找到任何有用的网站。任何帮助将不胜感激。
我的代码:
model.compile(loss='categorical_crossentropy', optimizer=opt_adam, metrics=['accuracy'])
history=model.fit_generator(generate_arrays_for_training(indexPat, train_data, start=0,end=100)
validation_data=generate_arrays_for_training(indexPat, test_data, start=0,end=100)
steps_per_epoch=int((len(train_data)/2)),
validation_steps=int((len(test_data)/2)),
verbose=2,class_weight = {0:1, 1:1.181},
epochs=65, max_queue_size=2, shuffle=True)
【问题讨论】:
标签: python machine-learning keras deep-learning cross-entropy