[SPARK-21793] Correct validateAndTransformSchema in GaussianMixture and AFTSurvivalRegression

• Type: Bug
• Status: Resolved
• Priority: Minor
• Resolution: Fixed
• Affects Version/s: 2.1.1, 2.2.0
• Fix Version/s: 2.1.2, 2.2.1, 2.3.0
• Component/s: MLlib
• Assignee: Cedric Pelvet
• Reporter: Cedric Pelvet
• Created: 20/Aug/17 10:04
• Updated: 20/Aug/17 10:07
• Resolved: 20/Aug/17 10:06

Description

From user sharp-pixel:

The line SchemaUtils.appendColumn(schema, $(predictionCol), IntegerType) did not modify the variable schema, hence only the last line had any effect. A temporary variable is used to correctly append the two columns predictionCol and probabilityCol.

Issue Links

• links to: GitHub issue