Joseph K. Bradley created SPARK-4460:
----------------------------------------

             Summary: RandomForest classification uses wrong threshold
                 Key: SPARK-4460
                 URL: https://issues.apache.org/jira/browse/SPARK-4460
             Project: Spark
          Issue Type: Bug
          Components: MLlib
    Affects Versions: 1.2.0
            Reporter: Joseph K. Bradley


RandomForest was modified to use WeightedEnsembleModel, but it needs to use a 
different threshold for prediction than GradientBoosting does.

Fix: WeightedEnsembleModel.scala:70 should use threshold 0.0 for boosting and 
0.5 for forests.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org

Reply via email to