Function run_model_param_check

tests/python/test_model_compatibility.py:16–26

(name: str, config: Dict[str, Any])

def run_model_param_check(name: str, config: Dict[str, Any]) -> None:
    assert config["learner"]["learner_model_param"]["num_feature"] == str(4)
    assert config["learner"]["learner_train_param"]["booster"] == "gbtree"

    booster = config["learner"]["gradient_booster"]
    assert booster["name"] == "gbtree"
    if name.find("1.0.0rc1") != -1:
        # There's no `num_parallel_tree` in the model parameter in 1.0 (it was a
        # configuration instead of a model parameter).
        return
    assert booster["gbtree_model_param"]["num_parallel_tree"] == str(gm.kForests)


def run_booster_check(booster: xgboost.Booster, name: str) -> None: