Классовый дисбаланс для двоичной классификации в кератах

Question

Я делаю бинарную классификацию. Мой y_train похож на [0,1,1,0,0]. Это одномерное. Я использую сигмоид в качестве активации в последнем слое, а также только 1 выходной нейрон в последнем слое. Я могу использовать class_weight.compute_class_weight, чтобы найти веса классов из y_train, а затем передать их параметру class_weight в функции keras.fit. Будет ли это работать?

Answer 1

Да, это должно работать.Вес класса - это карта, которая указывает относительный вес между классами и учитывается при взвешивании результатов выборки во время обучения.