search

Xtra-Computing / thundergbm

ThunderGBM: Fast GBDTs and Random Forests on GPUs
Apache License 2.0
689 stars 84 forks source link

How to visualize the ThunderGBM? #77

Open DarkoAlexander opened 2 years ago

DarkoAlexander commented 2 years ago

Is there a way to see what is under the hood of the random forest? Is there a way to get the parameters of a trained model? Or maybe you can get the individual decisions trees as a nested-if-statement?

Thanks alot!

Kurt-Liuhf commented 2 years ago

Hi @DarkoAlexander, currently feature of visualzation is not supported in ThunderGBM, and we are considering to add this feature in the future. Also, we will be glad if you can make a contribution.

DarkoAlexander commented 2 years ago

You say there is no feature of visualization. Is there also no way to get the parameters of the trained model? I want to get the numerical values for each node in each tree in the forest.

Kurt-Liuhf commented 2 years ago

Right, there is no way to get the parameters of the trained model (or numerical values of each node) in Python for now. Maybe you can refer to this and implement your own method.