From d3a516de13eb95d968407855bd92bba5c25db3ea Mon Sep 17 00:00:00 2001 From: louffy Date: Tue, 30 Oct 2018 12:03:49 +0800 Subject: [PATCH] Update Main.scala bug fix --- node2vec_spark/src/main/scala/com/navercorp/Main.scala | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/node2vec_spark/src/main/scala/com/navercorp/Main.scala b/node2vec_spark/src/main/scala/com/navercorp/Main.scala index f3494e5..9088022 100644 --- a/node2vec_spark/src/main/scala/com/navercorp/Main.scala +++ b/node2vec_spark/src/main/scala/com/navercorp/Main.scala @@ -39,6 +39,12 @@ object Main { opt[Int]("numWalks") .text(s"numWalks: ${defaultParams.numWalks}") .action((x, c) => c.copy(numWalks = x)) + opt[Int]("dim") + .text(s"dim: ${defaultParams.dim}") + .action((x, c) => c.copy(dim = x)) + opt[Int]("iter") + .text(s"iter: ${defaultParams.iter}") + .action((x, c) => c.copy(iter = x)) opt[Double]("p") .text(s"return parameter p: ${defaultParams.p}") .action((x, c) => c.copy(p = x))