From 74330086a5206e5fca8fea386b1604a6c540281e Mon Sep 17 00:00:00 2001
From: tangwang <tangwang@essa.top>
Date: Tue, 21 Oct 2025 22:56:01 +0800
Subject: [PATCH] deepwalk refactor for memsave and perfermance optimize

---
 offline_tasks/deepwalk/__init__.py | 8 ++++++++
 offline_tasks/deepwalk/deepwalk.py | 2 +-
 2 files changed, 9 insertions(+), 1 deletion(-)
 create mode 100644 offline_tasks/deepwalk/__init__.py

diff --git a/offline_tasks/deepwalk/__init__.py b/offline_tasks/deepwalk/__init__.py
new file mode 100644
index 0000000..8385722
--- /dev/null
+++ b/offline_tasks/deepwalk/__init__.py
@@ -0,0 +1,8 @@
+"""
+DeepWalk package for graph embedding
+"""
+from .deepwalk import DeepWalk
+from .alias import create_alias_table, alias_sample
+
+__all__ = ['DeepWalk', 'create_alias_table', 'alias_sample']
+
diff --git a/offline_tasks/deepwalk/deepwalk.py b/offline_tasks/deepwalk/deepwalk.py
index 7908620..14e6a51 100644
--- a/offline_tasks/deepwalk/deepwalk.py
+++ b/offline_tasks/deepwalk/deepwalk.py
@@ -3,7 +3,7 @@ import numpy as np
 import networkx as nx
 from joblib import Parallel, delayed
 import itertools
-from alias import create_alias_table, alias_sample
+from .alias import create_alias_table, alias_sample
 from tqdm import tqdm
 import argparse
 import multiprocessing
--
libgit2 0.21.2