Access to predictions of individual trees in random forest regression model (Python)

141 views
Skip to first unread message

Mohammadali Tarrahi

unread,
Jun 21, 2018, 2:21:04 PM6/21/18
to h2os...@googlegroups.com
Hi,

Hope you are doing well and thank you in advance for your help.
I am building H2O random forest regression model in Python, and I wonder how I can get the predictions of individual trees or if I can get the prediction quantiles?
I have hard time finding the solution both in the documentations and the web.

Thanks for your awesome package btw!
Many thanks,
Mohammadali

laurend

unread,
Jun 22, 2018, 3:38:35 PM6/22/18
to H2O Open Source Scalable Machine Learning - h2ostream
Hi Mohammadali,

Here is a tutorial that you might find helpful in answering your question: https://github.com/h2oai/h2o-tutorials/blob/master/tutorials/display_tree_mojo/DisplayTreeMojoH2O.ipynb.

cheers,

Lauren

Mohammadali Tarrahi

unread,
Jun 25, 2018, 9:36:53 AM6/25/18
to H2O Open Source Scalable Machine Learning - h2ostream
Hi Lauren,

Thanks a lot for the help.
I have used this tutorial to visualize individual trees but I am interested in each tree's prediction value,
interestingly H2O outputs "predict_leaf_node_assignment" but it does not output individual prediction values.

laurend

unread,
Jun 25, 2018, 2:53:14 PM6/25/18
to H2O Open Source Scalable Machine Learning - h2ostream
reposting gitter response to this question for other to see:

H2O doesn't currently have an API that would expose the individual predictions. There is a PR that adds this capability to MOJO, but it still needs to be finished https://github.com/h2oai/h2o-3/pull/1845/files

- lauren
Reply all
Reply to author
Forward
0 new messages