yu peng created SPARK-20418: ------------------------------- Summary: 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.3.15#6346) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org