-
-
Notifications
You must be signed in to change notification settings - Fork 8.7k
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
[jvm-packages] enable predictLeaf/predictContrib/treeLimit in 0.8 #3532
Conversation
Codecov Report
@@ Coverage Diff @@
## master #3532 +/- ##
============================================
+ Coverage 45.64% 45.86% +0.21%
Complexity 188 188
============================================
Files 166 166
Lines 13234 13320 +86
Branches 445 468 +23
============================================
+ Hits 6041 6109 +68
- Misses 6989 6994 +5
- Partials 204 217 +13
Continue to review full report at Codecov.
|
@CodingCat Should we shoot for merging this PR before releasing 0.80? If we need to, we can delay 0.80 release a bit. |
@hcho3 , yeah this is a blocking issue for the next release, too busy in daily jobs and also that tutorial thing, need more time (expectation is Friday) to add unit test cases here |
ef3316f
to
c342fa9
Compare
@yanboliang would you please take a look? |
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.
Looks pretty good overall, I left one minor comments.
predLeafItr: Iterator[Row], | ||
predContribItr: Iterator[Row]): Iterator[Row] = { | ||
// the following implementation is to be improved | ||
if (isDefined(leafPredictionCol) && isDefined(contribPredictionCol)) { |
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.
I have a little concern that users may set leafPredictionCol
with ""
, their intention should be not to output this column, but here we don't handle this case. At MLlib, we check $(leafPredictionCol).isEmpty
as well. What do you think about it?
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.
yes, good point, let me revise it
@yanboliang please check the latest update |
Looks good, it seams the failure is not relevant. |
I fixed the error in the Jenkins build. It should build successfully now. |
@hcho3 I merged this and cherry-picked to release_0.80 branch |
@CodingCat Okay, I will do a code freeze using this commit. The branch will be re-based against the master at this commit. As for the doc building, I'm going to place the build script on the Jenkins CI server. This way, we can install Java, Scala, and Maven. The built Javadoc will be placed to an S3 bucket, which will be then downloaded by ReadTheDocs. Expect a commit by end of today. |
Why Xgboost4j predictLeaf method is always have 3 leaves ? |
missing the test cases