[ https://issues.apache.org/jira/browse/SPARK-16739?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15625133#comment-15625133 ]
zhengruifeng commented on SPARK-16739: -------------------------------------- Agreed. Since GBDT is not a {{Classifier}}, we can not use it via {{OneVsRest}} to support multi-class classification. I think we need to refactor {{GBTClassifier}} to be a subclass of {{Classifier}}. [~josephkb] [~yanboliang] What's your opinions? {code} scala> val gbt = new GBTClassifier gbt: org.apache.spark.ml.classification.GBTClassifier = gbtc_506406adcf12 scala> val ovr = new OneVsRest().setClassifier(gbt) <console>:39: error: type mismatch; found : org.apache.spark.ml.classification.GBTClassifier required: org.apache.spark.ml.classification.Classifier[_, _, _] val ovr = new OneVsRest().setClassifier(gbt) ^ {code} > GBTClassifier should be a Classifier, not Predictor > --------------------------------------------------- > > Key: SPARK-16739 > URL: https://issues.apache.org/jira/browse/SPARK-16739 > Project: Spark > Issue Type: Improvement > Reporter: Vladimir Feinberg > Priority: Minor > > Should probably wait for SPARK-4240 to be resolved first. -- This message was sent by Atlassian JIRA (v6.3.4#6332) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org