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