[ 
https://issues.apache.org/jira/browse/SPARK-20418?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16102280#comment-16102280
 ] 

Weichen Xu commented on SPARK-20418:
------------------------------------

I will work on this.

> multi-label classification support
> ----------------------------------
>
>                 Key: SPARK-20418
>                 URL: https://issues.apache.org/jira/browse/SPARK-20418
>             Project: Spark
>          Issue Type: New Feature
>          Components: ML
>    Affects Versions: 2.1.0
>            Reporter: yu peng
>
> add multi-label label point and make a binary relevance based classifier 
> holder.
> http://scikit-learn.org/dev/modules/generated/sklearn.multioutput.MultiOutputClassifier.html#sklearn.multioutput.MultiOutputClassifier
> ```
> X = [ [ 0, 1, 0], [1, 0, 1]] 
> y = [[1], [1, 2]]
> df = sql.CreateDataframe(zip(X, y), ['x', 'y'])
> mlp_clf = 
> MultiOutputClassifier(LogisticRegression).setFeature('X').setLabel('y')
> mlp_clf.fit(df)
> ```



--
This message was sent by Atlassian JIRA
(v6.4.14#64029)

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

Reply via email to