[
https://issues.apache.org/jira/browse/SPARK-7424?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14531875#comment-14531875
]
Joseph K. Bradley commented on SPARK-7424:
------------------------------------------
I started a little work on this. It will involve modifying
PredictorParams.validateAndTransformSchema to copy metadata from the labelCol
to the outputCol, if available. It should not of course copy the column name.
The PredictionModel will need to store the labelCol attribute, if available.
This may require modifying subclasses. It may also require specializing
validateAndTransformSchema for Predictor and PredictionModel (making 2
versions).
> spark.ml classification, regression abstractions should add metadata to
> output column
> -------------------------------------------------------------------------------------
>
> Key: SPARK-7424
> URL: https://issues.apache.org/jira/browse/SPARK-7424
> Project: Spark
> Issue Type: Improvement
> Components: ML
> Reporter: Joseph K. Bradley
> Priority: Minor
>
> Update ClassificationModel, ProbabilisticClassificationModel prediction to
> include numClasses in output column metadata.
> Update RegressionModel to specify output column metadata as well.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
