06f2f82 replay all

apache · Dec 9, 2017 · beacf8a · beacf8a
1 parent 98a35d0
commit beacf8a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/spark/spark-2.0/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala b/spark/spark-2.0/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
@@ -221,15 +221,15 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) {
   }
 
   /**
-   * @see hivemall.evaluation.FMeasureUDAF
+   * @see hivemall.evaluation.F1ScoreUDAF
    * @group evaluation
    */
   def f1score(predict: String, target: String): DataFrame = {
     // checkType(target, ArrayType(IntegerType))
     // checkType(predict, ArrayType(IntegerType))
     val udaf = HiveUDAFFunction(
         "f1score",
-        new HiveFunctionWrapper("hivemall.evaluation.FMeasureUDAF"),
+        new HiveFunctionWrapper("hivemall.evaluation.F1ScoreUDAF"),
         Seq(predict, target).map(df.col(_).expr),
         isUDAFBridgeRequired = true)
       .toAggregateExpression()

diff --git a/spark/spark-2.2/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala b/spark/spark-2.2/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala
@@ -222,15 +222,15 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) {
   }
 
   /**
-   * @see hivemall.evaluation.FMeasureUDAF
+   * @see hivemall.evaluation.F1ScoreUDAF
    * @group evaluation
    */
   def f1score(predict: String, target: String): DataFrame = {
     // checkType(target, ArrayType(IntegerType))
     // checkType(predict, ArrayType(IntegerType))
     val udaf = HiveUDAFFunction(
         "f1score",
-        new HiveFunctionWrapper("hivemall.evaluation.FMeasureUDAF"),
+        new HiveFunctionWrapper("hivemall.evaluation.F1ScoreUDAF"),
         Seq(predict, target).map(df.col(_).expr),
         isUDAFBridgeRequired = true)
       .toAggregateExpression()