[SPARK-20043] Decision Tree loader does not handle uppercase impurity param values

• Type: Bug
• Status: Resolved
• Priority: Major
• Resolution: Fixed
• Affects Version/s: 2.1.0, 2.2.0
• Fix Version/s: 2.1.1, 2.2.0
• Component/s: ML
• Labels: starter
• Target Version/s: 2.1.1, 2.2.0
• Assignee: Yan Facai (颜发才)
• Reporter: Zied Sellami
• Shepherd: Joseph K. Bradley
• Created: 21/Mar/17 09:16
• Updated: 28/Mar/17 23:14
• Resolved: 28/Mar/17 23:14

Description

I saved a CrossValidatorModel with a decision tree and a random forest. I use Paramgrid to test "gini" and "entropy" impurity. CrossValidatorModel are not able to load the saved model, when impurity are written not in lowercase. I obtain an error from Spark "impurity Gini (Entropy) not recognized.

Issue Links

• links to: [Github] Pull Request #17407 (facaiy)