Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[python-package] Introduce refit_tree_manual to Booster class. #6617

Open
wants to merge 13 commits into
base: master
Choose a base branch
from
Prev Previous commit
Next Next commit
Fix for 3.8
  • Loading branch information
Atanas Dimitrov committed Aug 16, 2024
commit a387aa50291c470e9dc293b027f1f0bc265643b6
2 changes: 1 addition & 1 deletion tests/python_package_test/test_engine.py
Original file line number Diff line number Diff line change
@@ -2341,7 +2341,7 @@ def retrieve_leaves_from_tree(tree):
left_child = retrieve_leaves_from_tree(tree["left_child"])
right_child = retrieve_leaves_from_tree(tree["right_child"])

return left_child | right_child
return {**left_child, **right_child}

def retrieve_leaves_from_booster(booster, iteration):
tree = booster.dump_model(0, iteration)["tree_info"][0]["tree_structure"]
Loading
Oops, something went wrong.