зеркало из https://github.com/microsoft/LightGBM.git
fix unbalance for binary classification
This commit is contained in:
Родитель
47313fb574
Коммит
b3e99b27a0
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
|
@ -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;
|
||||
}
|
||||
}
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче