[ 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