From 62b173d7795eebeea5ac98bc86e7c9e10fa211fb Mon Sep 17 00:00:00 2001
From: ludatabricks <38018689+ludatabricks@users.noreply.github.com>
Date: Thu, 7 Jun 2018 13:21:32 -0700
Subject: [PATCH] Output Training Time as metrics (#148)

* change the structure of mlresult and add isLargerBetter

* output training time, not scoreTrainTime
---
 .../spark/sql/perf/mllib/MLPipelineStageBenchmarkable.scala     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/scala/com/databricks/spark/sql/perf/mllib/MLPipelineStageBenchmarkable.scala b/src/main/scala/com/databricks/spark/sql/perf/mllib/MLPipelineStageBenchmarkable.scala
index edb272d..32605bd 100644
--- a/src/main/scala/com/databricks/spark/sql/perf/mllib/MLPipelineStageBenchmarkable.scala
+++ b/src/main/scala/com/databricks/spark/sql/perf/mllib/MLPipelineStageBenchmarkable.scala
@@ -72,7 +72,7 @@ class MLPipelineStageBenchmarkable(
       val (scoreTrainTime, scoreTraining) = measureTime {
         test.score(param, trainingData, model)
       }
-      val metricTrainingTime = MLMetric("training.time", scoreTrainTime.toMillis, false)
+      val metricTrainingTime = MLMetric("training.time", trainingTime.toMillis, false)
       val metricTraining = MLMetric("training."+scoreTraining.metricName,
         scoreTraining.metricValue,
         scoreTraining.isLargerBetter)