add log

10d6fed4 · 高雅喆 · 1423c987 · 10d6fed4
Commit 10d6fed4 authored Oct 02, 2018 by 高雅喆
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

Main.scala eda/node2vec/src/main/scala/com/gmei/Main.scala +4 -0

No files found.
--- a/eda/node2vec/src/main/scala/com/gmei/Main.scala
+++ b/eda/node2vec/src/main/scala/com/gmei/Main.scala
@@ -170,6 +170,10 @@ object Main {
      import sc.implicits._
      // remap both ids back to words
      val remapFirst = similarityMatrix.entries.keyBy(_.i).join(index).values
+
+      println("get remapFirst")
+      println(remapFirst.first())
+
      val remapSecond = remapFirst.keyBy { case (entry, word1) => entry.j }.join(index).values.map {
        case ((entry, word1), word2) =>
          (word1, word2, entry.value)