[SPARK-12367] NoSuchElementException during prediction with Random Forest Regressor

• Type: Bug
• Status: Closed
• Priority: Major
• Resolution: Duplicate
• Affects Version/s: 1.5.2
• Component/s: ML
• Assignee: Unassigned
• Reporter: Eugene Morozov
• Created: 16/Dec/15 12:06
• Updated: 16/Dec/15 20:02
• Resolved: 16/Dec/15 20:02

Description

I'm consistently getting "java.util.NoSuchElementException: key not found: 1.0" while trying to do a prediction on a trained model.
I use ml package - Pipeline API. The model is successfully trained, I see some stats in the output: total, findSplitsBins, findBestSplits, chooseSplits. I can even serialize it into a file and use afterwards, but the prediction is broken somehow.

Code, input data and stack trace attached.

Issue Links

• duplicates: SPARK-12375 VectorIndexer: allow unknown categories