[SPARK-21476] RandomForest classification model not using broadcast in transform

• Type: Improvement
• Status: Resolved
• Priority: Minor
• Resolution: Incomplete
• Affects Version/s: 2.2.0
• Component/s: ML
• Labels: bulk-closed
• Assignee: Unassigned
• Reporter: Saurabh Agrawal
• Created: 19/Jul/17 19:52
• Updated: 21/May/19 04:17
• Resolved: 21/May/19 04:17

Description

I notice significant task deserialization latency while running prediction with pipelines using RandomForestClassificationModel. While digging into the source, found that the transform method in RandomForestClassificationModel binds to its parent ProbabilisticClassificationModel and the only concrete definition that RandomForestClassificationModel provides and which is actually used in transform is that of predictRaw. Broadcasting is not being used in predictRaw.