Repository: incubator-hivemall Updated Branches: refs/heads/master c1cd4b2e0 -> 5cb3c1666
Fixed CI error about f1score in Spark Project: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/commit/5cb3c166 Tree: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/tree/5cb3c166 Diff: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/diff/5cb3c166 Branch: refs/heads/master Commit: 5cb3c16663c8dde557d7d432c5944a3a6ac72eee Parents: c1cd4b2 Author: Makoto Yui <[email protected]> Authored: Thu Sep 14 14:21:33 2017 +0900 Committer: Makoto Yui <[email protected]> Committed: Thu Sep 14 14:21:33 2017 +0900 ---------------------------------------------------------------------- .../scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-hivemall/blob/5cb3c166/spark/spark-2.1/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala ---------------------------------------------------------------------- diff --git a/spark/spark-2.1/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala b/spark/spark-2.1/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala index a68f88f..7d2cd83 100644 --- a/spark/spark-2.1/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala +++ b/spark/spark-2.1/src/main/scala/org/apache/spark/sql/hive/HivemallGroupedDataset.scala @@ -221,7 +221,7 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) { } /** - * @see hivemall.evaluation.FMeasureUDAF + * @see hivemall.evaluation.F1ScoreUDAF * @group evaluation */ def f1score(predict: String, target: String): DataFrame = { @@ -229,7 +229,7 @@ final class HivemallGroupedDataset(groupBy: RelationalGroupedDataset) { // 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()
