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

["Feynman Liang (JIRA)" <ji...@apache.org>]

Feynman Liang created SPARK-8799:
------------------------------------

             Summary: 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
(v6.3.4#6332)

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