[jira] [Resolved] (SPARK-58994) Fix prediction column metadata in GBTRegressionModel

["Ruifeng Zheng (Jira)" <ji...@apache.org>]

     [ https://issues.apache.org/jira/browse/SPARK-58994?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]

Ruifeng Zheng resolved SPARK-58994.
-----------------------------------
    Fix Version/s: 4.0.5
                   4.1.4
                   4.2.1
                   4.3.0
       Resolution: Fixed

Issue resolved by pull request 58277
[https://github.com/apache/spark/pull/58277]

> Fix prediction column metadata in GBTRegressionModel
> ----------------------------------------------------
>
>                 Key: SPARK-58994
>                 URL: https://issues.apache.org/jira/browse/SPARK-58994
>             Project: Spark
>          Issue Type: Bug
>          Components: ML
>    Affects Versions: 5.0.0
>            Reporter: Ruifeng Zheng
>            Assignee: Ruifeng Zheng
>            Priority: Major
>              Labels: pull-request-available
>             Fix For: 4.0.5, 4.1.4, 4.2.1, 4.3.0
>
>
> GBTRegressionModel.transform aliases its prediction expression with featuresCol and the feature
> column metadata. Although withColumns assigns the final prediction column name, its outer alias
> inherits the feature metadata from the inner alias. The transformed prediction column can therefore
> carry vector feature metadata instead of the numeric prediction metadata produced by
> transformSchema.
> The fix uses predictionCol and its output metadata for the prediction expression and adds a
> regression test comparing the transformed metadata with transformSchema.
> Pull request: https://github.com/apache/spark/pull/58277



--
This message was sent by Atlassian Jira
(v8.20.10#820010)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org