From 39f0b386cbcbe2e54da01f4655b9dbf970d53ca9 Mon Sep 17 00:00:00 2001 From: Anahita Bhiwandiwalla Date: Thu, 8 Oct 2015 12:43:10 -0700 Subject: [PATCH] Fix for Kmeans doc bug --- .../engine/model/plugins/clustering/KMeansPredictPlugin.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/engine-plugins/model-plugins/src/main/scala/org/trustedanalytics/atk/engine/model/plugins/clustering/KMeansPredictPlugin.scala b/engine-plugins/model-plugins/src/main/scala/org/trustedanalytics/atk/engine/model/plugins/clustering/KMeansPredictPlugin.scala index acc44a5381..6a644ffcc4 100644 --- a/engine-plugins/model-plugins/src/main/scala/org/trustedanalytics/atk/engine/model/plugins/clustering/KMeansPredictPlugin.scala +++ b/engine-plugins/model-plugins/src/main/scala/org/trustedanalytics/atk/engine/model/plugins/clustering/KMeansPredictPlugin.scala @@ -111,11 +111,11 @@ class KMeansPredictPlugin extends SparkCommandPlugin[KMeansPredictArgs, FrameRef var columnNames = new ListBuffer[String]() var columnTypes = new ListBuffer[DataTypes.DataType]() for (i <- 1 to kmeansModel.clusterCenters.length) { - val colName = "distance_from_cluster_" + i.toString + val colName = "Distance_from_Cluster:" + i.toString columnNames += colName columnTypes += DataTypes.float64 } - columnNames += "predicted_cluster" + columnNames += "Predicted_Cluster" columnTypes += DataTypes.int32 val newColumns = columnNames.toList.zip(columnTypes.toList.map(x => x: DataType))