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

[docs] add supertree in README #6625

Merged
merged 3 commits into from
Sep 3, 2024
Merged

[docs] add supertree in README #6625

merged 3 commits into from
Sep 3, 2024

Conversation

pplonski
Copy link
Contributor

Hi LightGBM Team!

I've added in readme a new package for decision tree visualization - supertree. It works with LGBMClassifier and LGBMRegressor. The package can be used in notebook environment (Jupyter, Colab) or in Python script. I hope it will be useful for LightGBM users. Thank you!

Copy link
Collaborator

@jameslamb jameslamb left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

sure! Thanks

@jameslamb jameslamb changed the title add supertree in README [docs] add supertree in README Aug 28, 2024
@StrikerRUS
Copy link
Collaborator

It works with LGBMClassifier and LGBMRegressor.

Looks like this package can visualize Booster as well. 👍

image

https://github.com/mljar/supertree/blob/7a0b90e0414705a8570dc05eaf83861ead5e6729/supertree/supertree.py#L487-L492

Copy link
Collaborator

@StrikerRUS StrikerRUS left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for your contribution!

@jameslamb jameslamb merged commit b1203c9 into microsoft:master Sep 3, 2024
44 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants