thundergbm icon indicating copy to clipboard operation
thundergbm copied to clipboard

How to visualize the ThunderGBM?

Open DarkoAlexander opened this issue 2 years ago • 3 comments

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!

DarkoAlexander avatar Mar 07 '22 12:03 DarkoAlexander

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.

Kurt-Liuhf avatar Mar 09 '22 06:03 Kurt-Liuhf

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.

DarkoAlexander avatar Mar 09 '22 17:03 DarkoAlexander

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.

Kurt-Liuhf avatar Mar 10 '22 07:03 Kurt-Liuhf