[
https://issues.apache.org/jira/browse/SPARK-8799?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Joseph K. Bradley updated SPARK-8799:
-------------------------------------
Target Version/s: 3.0.0
> OneVsRestModel should extend ClassificationModel
> ------------------------------------------------
>
> Key: SPARK-8799
> URL: https://issues.apache.org/jira/browse/SPARK-8799
> Project: Spark
> Issue Type: Improvement
> Components: ML
> Reporter: Feynman Liang
> Priority: Minor
>
> Many parts of `OneVsRestModel` can be generalized to `ClassificationModel`.
> For example:
> * `accColName` can be used to populate `ClassificationModel#predictRaw` and
> share implementations of `transform`
> * SPARK-8092 adds `setFeaturesCol` and `setPredictionCol` which could be
> gotten for free through subclassing
> `ClassificationModel` is the correct supertype (e.g. not `PredictionModel`)
> because the labels for a `OneVsRest` will always be discrete and finite.
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
