log quantileProbabilities.size

apache · Jan 13, 2017 · 7f00d36 · 7f00d36
1 parent e6b4615
commit 7f00d36
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala b/mllib/src/main/scala/org/apache/spark/ml/regression/AFTSurvivalRegression.scala
@@ -229,7 +229,8 @@ class AFTSurvivalRegression @Since("1.6.0") (@Since("1.6.0") override val uid: S
 
     val instr = Instrumentation.create(this, dataset)
     instr.logParams(labelCol, featuresCol, censorCol, predictionCol, quantilesCol,
-      quantileProbabilities, fitIntercept, maxIter, tol, aggregationDepth)
+      fitIntercept, maxIter, tol, aggregationDepth)
+    instr.logNamedValue("quantileProbabilities.size", $(quantileProbabilities).length)
     instr.logNumFeatures(numFeatures)
 
     if (!$(fitIntercept) && (0 until numFeatures).exists { i =>