You are viewing a plain text version of this content. The canonical link for it is here.
Posted to issues@spark.apache.org by "Joseph Wang (JIRA)" <ji...@apache.org> on 2017/08/01 14:30:01 UTC

[jira] [Created] (SPARK-21594) Probability output from MutilayerPerceptronClassifier

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-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org