Skip to content
Snippets Groups Projects
Commit 2f739567 authored by Bryan Cutler's avatar Bryan Cutler Committed by Joseph K. Bradley
Browse files

[SPARK-17697][ML] Fixed bug in summary calculations that pattern match against...

[SPARK-17697][ML] Fixed bug in summary calculations that pattern match against label without casting

## What changes were proposed in this pull request?
In calling LogisticRegression.evaluate and GeneralizedLinearRegression.evaluate using a Dataset where the Label is not of a double type, calculations pattern match against a double and throw a MatchError.  This fix casts the Label column to a DoubleType to ensure there is no MatchError.

## How was this patch tested?
Added unit tests to call evaluate with a dataset that has Label as other numeric types.

Author: Bryan Cutler <cutlerb@gmail.com>

Closes #15288 from BryanCutler/binaryLOR-numericCheck-SPARK-17697.
parent 39eb3bb1
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment