From d184e888b615470f9acc9edd268ac16554e42140 Mon Sep 17 00:00:00 2001 From: jyswpp Date: Tue, 16 Jan 2024 15:17:03 +0800 Subject: [PATCH] fix some bug for deepwalk --- .../tencent/angel/spark/examples/cluster/DeepWalkExample.scala | 1 + .../embedding/deepwalkNoWeight/DeepWalkGraphPartition.scala | 1 + 2 files changed, 2 insertions(+) diff --git a/spark-on-angel/examples/src/main/scala/com/tencent/angel/spark/examples/cluster/DeepWalkExample.scala b/spark-on-angel/examples/src/main/scala/com/tencent/angel/spark/examples/cluster/DeepWalkExample.scala index 817cf18b1..447d9d510 100644 --- a/spark-on-angel/examples/src/main/scala/com/tencent/angel/spark/examples/cluster/DeepWalkExample.scala +++ b/spark-on-angel/examples/src/main/scala/com/tencent/angel/spark/examples/cluster/DeepWalkExample.scala @@ -20,6 +20,7 @@ package com.tencent.angel.spark.examples.cluster import com.tencent.angel.spark.context.PSContext import com.tencent.angel.spark.ml.core.ArgsUtil import com.tencent.angel.graph.embedding.deepwalk.DeepWalk +import com.tencent.angel.graph.embedding.deepwalkNoWeight.DeepWalkNoWeight import com.tencent.angel.graph.utils.GraphIO import org.apache.spark.storage.StorageLevel import org.apache.spark.{SparkConf, SparkContext} diff --git a/spark-on-angel/graph/src/main/scala/com/tencent/angel/graph/embedding/deepwalkNoWeight/DeepWalkGraphPartition.scala b/spark-on-angel/graph/src/main/scala/com/tencent/angel/graph/embedding/deepwalkNoWeight/DeepWalkGraphPartition.scala index 6655b7339..8b3ea5fd5 100644 --- a/spark-on-angel/graph/src/main/scala/com/tencent/angel/graph/embedding/deepwalkNoWeight/DeepWalkGraphPartition.scala +++ b/spark-on-angel/graph/src/main/scala/com/tencent/angel/graph/embedding/deepwalkNoWeight/DeepWalkGraphPartition.scala @@ -1,5 +1,6 @@ package com.tencent.angel.graph.embedding.deepwalkNoWeight +import com.tencent.angel.graph.psf.neighbors.SampleNeighborsWithCount.NeighborsTableElement import it.unimi.dsi.fastutil.longs.Long2ObjectOpenHashMap import scala.collection.mutable.ArrayBuffer