Get individual tree depths

[abhishek-ghose]

Hi,

I am using the Python version of LightGBM (not the sklearn API though). After I have run the train() function with boosting_type='gbdt', is there anyway to get the depths of individual trees? I have the prescribed max_depths; I am interested in the actual depths.

I see that this information is printed out per iteration:
[LightGBM] [Debug] Trained a tree with leaves = 31 and max_depth = 8

But I am unable to locate this information in the object train() returns (checked it in debug mode). Is this information stored only in the C backend? Any means to access it from Python?

Thanks!

[guolinke]

set verbose=2

[abhishek-ghose]

Hi, what attribute of the Python model should I use to access the tree depths? To be clear, I already see it printed - that's not my question; I want to access the depths of each of the trees in the gbm model within my Python code. Is there a way to do that?

[guolinke]

@abhishek-ghose
sorry, the depth is not stored in the model content.
I think the simplest solution for your question is, redirecting the log into a file, and get the depths from that.

[abhishek-ghose]

Got it, thanks!