In 2-class problems, when I use SVM, RondomForest models to do classifications, they predict "0" or "1". And when I use ROC to evaluate the model, sometimes I need a probability that a record belongs to "0" or "1". In scikit-learn, every model can do "predict" and "predict_prob", which the last one can ouput the probability. I find the document, and didn't found this function in MLLIB. Does mllib has this function?
-- View this message in context: http://apache-spark-user-list.1001560.n3.nabble.com/Mllib-using-model-to-predict-probability-tp26886.html Sent from the Apache Spark User List mailing list archive at Nabble.com. --------------------------------------------------------------------- To unsubscribe, e-mail: user-unsubscr...@spark.apache.org For additional commands, e-mail: user-h...@spark.apache.org
