From a7d8af9f904d7a5d07fe90899379d8ff555e1d8a Mon Sep 17 00:00:00 2001 From: Rahul Tanwani Date: Tue, 16 Feb 2016 14:54:16 +0000 Subject: [PATCH] [MINOR][MLLIB] Public visibility for eval metric's dataframe constructor --- .../spark/mllib/evaluation/BinaryClassificationMetrics.scala | 2 +- .../org/apache/spark/mllib/evaluation/MulticlassMetrics.scala | 2 +- .../org/apache/spark/mllib/evaluation/RegressionMetrics.scala | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/evaluation/BinaryClassificationMetrics.scala b/mllib/src/main/scala/org/apache/spark/mllib/evaluation/BinaryClassificationMetrics.scala index 12cf22095720a..fbaf8e1c3fbee 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/evaluation/BinaryClassificationMetrics.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/evaluation/BinaryClassificationMetrics.scala @@ -57,7 +57,7 @@ class BinaryClassificationMetrics @Since("1.3.0") ( * An auxiliary constructor taking a DataFrame. * @param scoreAndLabels a DataFrame with two double columns: score and label */ - private[mllib] def this(scoreAndLabels: DataFrame) = + def this(scoreAndLabels: DataFrame) = this(scoreAndLabels.map(r => (r.getDouble(0), r.getDouble(1)))) /** diff --git a/mllib/src/main/scala/org/apache/spark/mllib/evaluation/MulticlassMetrics.scala b/mllib/src/main/scala/org/apache/spark/mllib/evaluation/MulticlassMetrics.scala index c5104960cfcb6..6df3d992c2ec2 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/evaluation/MulticlassMetrics.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/evaluation/MulticlassMetrics.scala @@ -37,7 +37,7 @@ class MulticlassMetrics @Since("1.1.0") (predictionAndLabels: RDD[(Double, Doubl * An auxiliary constructor taking a DataFrame. * @param predictionAndLabels a DataFrame with two double columns: prediction and label */ - private[mllib] def this(predictionAndLabels: DataFrame) = + def this(predictionAndLabels: DataFrame) = this(predictionAndLabels.map(r => (r.getDouble(0), r.getDouble(1)))) private lazy val labelCountByClass: Map[Double, Long] = predictionAndLabels.values.countByValue() diff --git a/mllib/src/main/scala/org/apache/spark/mllib/evaluation/RegressionMetrics.scala b/mllib/src/main/scala/org/apache/spark/mllib/evaluation/RegressionMetrics.scala index 18c90b204a26a..dea19e677ba63 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/evaluation/RegressionMetrics.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/evaluation/RegressionMetrics.scala @@ -45,7 +45,7 @@ class RegressionMetrics @Since("2.0.0") ( * @param predictionAndObservations a DataFrame with two double columns: * prediction and observation */ - private[mllib] def this(predictionAndObservations: DataFrame) = + def this(predictionAndObservations: DataFrame) = this(predictionAndObservations.map(r => (r.getDouble(0), r.getDouble(1)))) /**