[SPARK-4460] RandomForest classification uses wrong threshold

• Type: Bug
• Status: Closed
• Priority: Major
• Resolution: Invalid
• Affects Version/s: 1.2.0
• Component/s: MLlib
• Target Version/s: 1.2.0
• Assignee: Unassigned
• Reporter: Joseph K. Bradley
• Created: 17/Nov/14 21:55
• Updated: 17/Nov/14 23:32
• Resolved: 17/Nov/14 23:32

Description

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.