Aman Rawat created SPARK-21416:
----------------------------------
Summary: Emit Raw Prediction Scores in DecisionTree Classification
Model
Key: SPARK-21416
URL: https://issues.apache.org/jira/browse/SPARK-21416
Project: Spark
Issue Type: Improvement
Components: ML
Affects Versions: 2.0.1
Reporter: Aman Rawat
DecisionTreeClassificationModel today emits only the final prediction score
(index for the class). However, it's often required to know the raw prediction
score for each class as well. This is already available in the underlying
model, but not exposed in "ML" (i.e. DataFrame based) interace -
DecisionTreeClassificationModel.
The transformer's output can be changed to include both "prediction" and
"rawPrediction" columns.
--
This message was sent by Atlassian JIRA
(v6.4.14#64029)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
