You are viewing a plain text version of this content. The canonical link for it is here.
Posted to issues@spark.apache.org by "Joseph K. Bradley (JIRA)" <ji...@apache.org> on 2018/06/13 22:59:01 UTC

[jira] [Updated] (SPARK-8799) OneVsRestModel should extend ClassificationModel

     [ 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:   (was: 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-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org