[SPARK-26458] OneHotEncoderModel verifies the number of category values incorrectly when tries to transform a dataframe.

• Type: Bug
• Status: Resolved
• Priority: Major
• Resolution: Not A Problem
• Affects Version/s: 2.3.1
• Component/s: ML
• Assignee: Unassigned
• Reporter: duruihuan
• Created: 28/Dec/18 03:39
• Updated: 01/Mar/19 23:16
• Resolved: 01/Mar/19 23:16

Description

When the handleInvalid is set to "keep", then one should not compare the categorySizes of the tranformSchema and the values of the metadata of the dataframe to be transformed. Because there may be more than one invalid values in some columns in the dataframe, which causes exception as described in lines 302-306 in OneHotEncoderEstimator.scala. To be concluded, I think the verifyNumOfValues in the method transformSchema should be removed, which can be found in line 299 in the code.