1

我正在研究一个不平衡的数据集。我正在尝试在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)

4

1 回答 1

2

您可以为此雇用compute_class_weightfrom sklearn。

from sklearn.utils import compute_class_weight
class_weights = compute_class_weight("balanced", np.unique(Y_train), Y_train)
于 2019-11-29T10:03:46.413 回答