keras中的类权重属性用于类平衡

时间:2019-03-02 06:04:34

标签: python keras neural-network nlp data-science

请记住,y是一个热编码

import numpy as np
from sklearn.utils.class_weight import compute_class_weight

y_integers = np.argmax(y, axis=1)
class_weights = compute_class_weight('balanced', np.unique(y_integers), y_integers)
d_class_weights = dict(enumerate(class_weights))

keras模型如何将这些class_weights映射到我们输入到模型中的one_hot_encoded标签?我想自定义二进制熵损失函数以实现类平衡。

0 个答案:

没有答案