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
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
def test_check_model_bounds_no_checks(capsys): hyper.check_model_bounds(BASE_PARAMS, BASE_PARAMS) captured = capsys.readouterr() assert '' == captured.out