github/spark-sql-perf
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Output Training Time as metrics (#148)

Browse Source 
* change the structure of mlresult and add isLargerBetter

* output training time, not scoreTrainTime
This commit is contained in:
ludatabricks 2018-06-07 13:21:32 -07:00 committed by Xiangrui Meng
parent d9984e1c0a
commit 62b173d779
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/scala/com/databricks/spark/sql/perf/mllib/MLPipelineStageBenchmarkable.scala
View File

@ -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)