Python BaseTuner.higher_is_betterの例

コード例 #1

ファイル: test_optimize.py プロジェクト: yuanjames/optuna

    def test_higher_is_better(self):
        # type: () -> None

        for metric in ['auc', 'accuracy']:
            tuner = BaseTuner(lgbm_params={'metric': metric})
            assert tuner.higher_is_better()

        for metric in ['rmsle', 'rmse', 'binary_logloss']:
            tuner = BaseTuner(lgbm_params={'metric': metric})
            assert not tuner.higher_is_better()

コード例 #2

    def test_higher_is_better(self):
        # type: () -> None

        for metric in [
                'auc', 'ndcg', 'lambdarank', 'rank_xendcg', 'xendcg',
                'xe_ndcg', 'xe_ndcg_mart', 'xendcg_mart', 'map',
                'mean_average_precision'
        ]:
            tuner = BaseTuner(lgbm_params={'metric': metric})
            assert tuner.higher_is_better()

        for metric in ['rmsle', 'rmse', 'binary_logloss']:
            tuner = BaseTuner(lgbm_params={'metric': metric})
            assert not tuner.higher_is_better()

コード例 #3

    def test_higher_is_better(self):
        # type: () -> None

        for metric in [
            "auc",
            "ndcg",
            "lambdarank",
            "rank_xendcg",
            "xendcg",
            "xe_ndcg",
            "xe_ndcg_mart",
            "xendcg_mart",
            "map",
            "mean_average_precision",
        ]:
            tuner = BaseTuner(lgbm_params={"metric": metric})
            assert tuner.higher_is_better()

        for metric in ["rmsle", "rmse", "binary_logloss"]:
            tuner = BaseTuner(lgbm_params={"metric": metric})
            assert not tuner.higher_is_better()