Python check_model_bounds示例

示例#1

文件： test_hyper.py 项目： sergshabal/poptimizer_old

def test_check_model_lower_bound(capsys):
    space = {
        'model': {
            'learning_rate': 0.1 / (1 + 0.91 * hyper.LEARNING_RATE_RANGE),
            'depth': 1,
            'l2_leaf_reg': 3 / (1 + 0.91 * hyper.L2_RANGE),
            'random_strength': 1 / (1 + 0.91 * hyper.RAND_STRENGTH_RANGE),
            'bagging_temperature': 1 / (1 + 0.91 * hyper.BAGGING_RANGE)
        }
    }
    hyper.check_model_bounds(space, BASE_PARAMS)
    captured = capsys.readouterr()
    assert 'Необходимо увеличить LEARNING_RATE_RANGE' in captured.out
    assert 'Необходимо увеличить L2_RANGE' in captured.out
    assert 'Необходимо увеличить RAND_STRENGTH_RANGE' in captured.out
    assert 'Необходимо увеличить BAGGING_RANGE' in captured.out

示例#2

文件： test_hyper.py 项目： sergshabal/poptimizer_old

def test_check_model_upper_bound(capsys):
    params = {
        'model': {
            'learning_rate': 0.1 * (1 + 0.91 * hyper.LEARNING_RATE_RANGE),
            'depth': hyper.MAX_DEPTH,
            'l2_leaf_reg': 3 * (1 + 0.91 * hyper.L2_RANGE),
            'random_strength': 1 * (1 + 0.91 * hyper.RAND_STRENGTH_RANGE),
            'bagging_temperature': 1 * (1 + 0.91 * hyper.BAGGING_RANGE)
        }
    }
    hyper.check_model_bounds(params, BASE_PARAMS)
    captured = capsys.readouterr()
    assert 'Необходимо увеличить LEARNING_RATE_RANGE' in captured.out
    assert 'Необходимо увеличить MAX_DEPTH' in captured.out
    assert 'Необходимо увеличить L2_RANGE' in captured.out
    assert 'Необходимо увеличить RAND_STRENGTH_RANGE' in captured.out
    assert 'Необходимо увеличить BAGGING_RANGE' in captured.out

示例#3

文件： test_hyper.py 项目： sergshabal/poptimizer_old

def test_check_model_bounds_no_checks(capsys):
    hyper.check_model_bounds(BASE_PARAMS, BASE_PARAMS)
    captured = capsys.readouterr()
    assert '' == captured.out