Joseph Wang created SPARK-21594: ----------------------------------- Summary: Probability output from MutilayerPerceptronClassifier Key: SPARK-21594 URL: https://issues.apache.org/jira/browse/SPARK-21594 Project: Spark Issue Type: New Feature Components: ML Affects Versions: 2.2.0 Environment: SPARK, PySpark,Scala, SparkR Reporter: Joseph Wang
My question is, is it possible to get not only the labels, but also (or only) the probability for that label? Like not just 0 or 1 for every input, but something like 0.95 for 0 and 0.05 for 1. If this is not possible with MLP, but is possible with other classifier. I have only used MLP because I know they should be capable of returning the probability, but I can't find it in PySpark. This is an inconsistent offering which needs to be fixed, which is provided by other algorithms in Spark MLlib with Spark Data Frame but not MLP which is related to AI stuff. thanks Joseph -- 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