[SPARK-15892][ML] Backport correctly merging AFTAggregators to branch 1.6 #13725
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What changes were proposed in this pull request?
This PR backports #13619.
The original test added in branch-2.0 was failed in branch-1.6.
This seems because the behaviour was changed in 101663f. This was failure while calculating Euler's number which ends up with a infinity regardless of this path.
So, I brought the dataset from
AFTSurvivalRegressionExampleto make sure this is working and then wrote the test.I ran the test before/after creating empty partitions.
model.scalebecomes1.0with empty partitions and becames1.547without them.After this patch, this becomes always
1.547.How was this patch tested?
Unit test in
AFTSurvivalRegressionSuite.