From 75934457d75996be71ffd0d4b448497d656c0d40 Mon Sep 17 00:00:00 2001
From: Zheng RuiFeng <ruifengz@foxmail.com>
Date: Mon, 14 Nov 2016 19:42:00 +0000
Subject: [PATCH] [SPARK-11496][GRAPHX][FOLLOWUP] Add param checking for
runParallelPersonalizedPageRank
## What changes were proposed in this pull request?
add the param checking to keep in line with other algos
## How was this patch tested?
existing tests
Author: Zheng RuiFeng <ruifengz@foxmail.com>
Closes #15876 from zhengruifeng/param_check_runParallelPersonalizedPageRank.
---
.../main/scala/org/apache/spark/graphx/lib/PageRank.scala | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/graphx/src/main/scala/org/apache/spark/graphx/lib/PageRank.scala b/graphx/src/main/scala/org/apache/spark/graphx/lib/PageRank.scala
index f4b00757a8..c0c3c73463 100644
--- a/graphx/src/main/scala/org/apache/spark/graphx/lib/PageRank.scala
+++ b/graphx/src/main/scala/org/apache/spark/graphx/lib/PageRank.scala
@@ -185,6 +185,13 @@ object PageRank extends Logging {
def runParallelPersonalizedPageRank[VD: ClassTag, ED: ClassTag](graph: Graph[VD, ED],
numIter: Int, resetProb: Double = 0.15,
sources: Array[VertexId]): Graph[Vector, Double] = {
+ require(numIter > 0, s"Number of iterations must be greater than 0," +
+ s" but got ${numIter}")
+ require(resetProb >= 0 && resetProb <= 1, s"Random reset probability must belong" +
+ s" to [0, 1], but got ${resetProb}")
+ require(sources.nonEmpty, s"The list of sources must be non-empty," +
+ s" but got ${sources.mkString("[", ",", "]")}")
+
// TODO if one sources vertex id is outside of the int range
// we won't be able to store its activations in a sparse vector
val zero = Vectors.sparse(sources.size, List()).asBreeze
--
GitLab