fix unbalance for binary classification

This commit is contained in:
Guolin Ke 2016-11-01 15:05:04 +08:00
Родитель 47313fb574
Коммит b3e99b27a0
2 изменённых файлов: 1140 добавлений и 2 удалений

Разница между файлами не показана из-за своего большого размера Загрузить разницу

Просмотреть файл

@ -47,8 +47,8 @@ public:
label_weights_[1] = 1.0f;
// if using unbalance, change the labels weight
if (is_unbalance_) {
label_weights_[1] = 1.0f / cnt_positive;
label_weights_[0] = 1.0f / cnt_negative;
label_weights_[1] = 1.0f;
label_weights_[0] = static_cast<float>(cnt_positive) / cnt_negative;
}
}