-
Notifications
You must be signed in to change notification settings - Fork 2k
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
PUBDEV-8691 Single Decision Tree implementation #6447
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There is couple of stuff that should be removed or improved before it goes public.
IMHO when we remove the stuff that I mention, we can have it in the codebase and polish it in different PR. There is no public documentation - only power user can try it with
from h2o.estimatorsimport H2ODecisionTreeEstimator
since there is no public documentation yet.
I would also consider to put it Expeprimental for the first implementation as SVM is.
@Override
public BuilderVisibility builderVisibility() { return BuilderVisibility.Experimental; }
TODOs I see for the differents PRs:
- scalability benchmarks
- Refactor DT API - the purpose of DT is to have nicer API than DTree
- Documentation in sphinx
- jupyter demo
And please rebase on current master, there are some conflict mentioned by github.
# Conflicts: # h2o-algos/src/main/java/hex/api/RegisterAlgos.java # h2o-r/H2O_Load.R # h2o-r/h2o-DESCRIPTION.template # h2o-r/scripts/h2o-r-test-setup.R
// compute score for given point | ||
CompressedDT tree = DKV.getGet(_output._treeKey); | ||
DTPrediction prediction = tree.predictRowStartingFromNode(data, 0, ""); | ||
// System.out.println(prediction.ruleExplanation); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove sout
This pull request replaces the old one (#6182)
https://h2oai.atlassian.net/browse/PUBDEV-8691