Атрибут веса классов в кератах для балансировки классов - PullRequest
0 голосов
/ 02 марта 2019

Помня, что 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, которые мы вводим в модель?Я хочу настроить двоичную функцию потери энтропии для балансировки классов.

...