我正在处理不平衡的数据集。我正在尝试借助的模型class_weights。因此，我可以在什么基础上确定class_weights。
标签及其数量如下：
label Count
2 47213
3 2096
4 2021
1 737
0 176
那么我应该为class_weight变量赋予什么值：
model.fit(X_train, Y_train, nb_epoch=5, batch_size=32, class_weight=class_weight)
您可以使用compute_class_weight从sklearn该。
from sklearn.utils import compute_class_weight
class_weights = compute_class_weight("balanced", np.unique(Y_train), Y_train)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句