It appears as if the inheritance hierarchy doesn't allow GBTClassifiers to be
used as the binary classifier in a OneVsRest trainer. Is there a simple way
to use gradient-boosted trees for multiclass (not binary) problems?
Specifically, it complains that GBTClassifier doesn't inherit from
Classifier[_, _, _].
I'm using Spark 2.0.1:
val gbt = new GBTClassifier()
.setLabelCol("indexedLabel")
.setFeaturesCol("features")
.setMaxIter(10)
.setMaxDepth(10)
val ovr = new OneVsRest().
setClassifier(gbt)
fails saying
error: type mismatch;
found : org.apache.spark.ml.classification.GBTClassifier
required: org.apache.spark.ml.classification.Classifier[_, _, _]
setClassifier(gbt)
--
View this message in context:
http://apache-spark-user-list.1001560.n3.nabble.com/Spark-ML-Using-GBTClassifier-in-OneVsRest-tp27933.html
Sent from the Apache Spark User List mailing list archive at Nabble.com.
---------------------------------------------------------------------
To unsubscribe e-mail: user-unsubscr...@spark.apache.org
