def test_get_booster_best_score__using_valid_names_as_list(self):
# type: () -> None
unexpected_value = 0.5
expected_value = 1.0
class DummyBooster(object):
def __init__(self):
# type: () -> None
self.best_score = {
"train": {"binary_logloss": unexpected_value},
"val": {"binary_logloss": expected_value},
}
booster = DummyBooster()
dummy_train_dataset = lgb.Dataset(None)
dummy_val_dataset = lgb.Dataset(None)
tuner = BaseTuner(
lgbm_kwargs={
"valid_names": ["train", "val"],
"valid_sets": [dummy_train_dataset, dummy_val_dataset],
}
)
val_score = tuner._get_booster_best_score(booster)
assert val_score == expected_value
def test_get_booster_best_score__using_valid_names_as_list(self):
# type: () -> None
unexpected_value = 0.5
expected_value = 1.0
class DummyBooster(object):
def __init__(self):
# type: () -> None
self.best_score = {
'train': {
'binary_logloss': unexpected_value
},
'val': {
'binary_logloss': expected_value
}
}
booster = DummyBooster()
dummy_train_dataset = lgb.Dataset(None)
dummy_val_dataset = lgb.Dataset(None)
tuner = BaseTuner(
lgbm_kwargs={
'valid_names': ['train', 'val'],
'valid_sets': [dummy_train_dataset, dummy_val_dataset],
})
val_score = tuner._get_booster_best_score(booster)
assert val_score == expected_value
def test_get_booster_best_score__using_valid_names_as_str(self):
# type: () -> None
expected_value = 1.0
class DummyBooster(object):
def __init__(self):
# type: () -> None
self.best_score = {"dev": {"binary_logloss": expected_value}}
booster = DummyBooster()
dummy_dataset = lgb.Dataset(None)
tuner = BaseTuner(lgbm_kwargs={"valid_names": "dev", "valid_sets": dummy_dataset,})
val_score = tuner._get_booster_best_score(booster)
assert val_score == expected_value
def test_get_booster_best_score(self):
# type: () -> None
expected_value = 1.0
class DummyBooster(object):
def __init__(self):
# type: () -> None
self.best_score = {"valid_0": {"binary_logloss": expected_value}}
booster = DummyBooster()
dummy_dataset = lgb.Dataset(None)
tuner = BaseTuner(lgbm_kwargs=dict(valid_sets=dummy_dataset))
val_score = tuner._get_booster_best_score(booster)
assert val_score == expected_value