def get_num_trees(booster: xgb.Booster) -> int:
data = [json.loads(d) for d in booster.get_dump(dump_format="json")]
return len(data)
def booster_text(booster: xgboost.Booster, base_score: float):
return increase_leaves_booster_text('\n'.join(booster.get_dump()),
base_score)
def tree_obj(bst: xgb.Booster):
return [json.loads(j) for j in bst.get_dump(dump_format="json")]
def get_num_trees(bst: xgb.Booster):
import json
data = [json.loads(d) for d in bst.get_dump(dump_format="json")]
return len(data) // 4
def _model_iteration(self, model: xgboost.Booster) -> int:
if not hasattr(model, "num_boosted_rounds"):
# Compatibility with XGBoost < 1.4
return len(model.get_dump())
return model.num_boosted_rounds()
