-
- Downloads
[MINOR] Correct validateAndTransformSchema in GaussianMixture and AFTSurvivalRegression
## What changes were proposed in this pull request? 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. ## How was this patch tested? Manually. Please review http://spark.apache.org/contributing.html before opening a pull request. Author: Cédric Pelvet <cedric.pelvet@gmail.com> Closes #18980 from sharp-pixel/master.
Showing
- mllib/src/main/scala/org/apache/spark/ml/clustering/GaussianMixture.scala 2 additions, 2 deletions...cala/org/apache/spark/ml/clustering/GaussianMixture.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala 5 additions, 3 deletions...rg/apache/spark/ml/regression/AFTSurvivalRegression.scala
Loading
Please register or sign in to comment