tangwang / recommendation

Commit 74330086a5206e5fca8fea386b1604a6c540281e

Authored by tangwang
1 parent e89d7a84

deepwalk refactor for memsave and perfermance optimize

Showing 2 changed files with 9 additions and 1 deletions   Show diff stats
Inline Side-by-side
offline_tasks/deepwalk/__init__.py 0 → 100644
  Show/Hide comments   View file @7433008
... ... @@ -0,0 +1,8 @@
  1 +"""
  2 +DeepWalk package for graph embedding
  3 +"""
  4 +from .deepwalk import DeepWalk
  5 +from .alias import create_alias_table, alias_sample
  6 +
  7 +__all__ = ['DeepWalk', 'create_alias_table', 'alias_sample']
  8 +
... ...
offline_tasks/deepwalk/deepwalk.py
  Show/Hide comments   View file @7433008
... ... @@ -3,7 +3,7 @@ import numpy as np
3 3 import networkx as nx
4 4 from joblib import Parallel, delayed
5 5 import itertools
6   -from alias import create_alias_table, alias_sample
  6 +from .alias import create_alias_table, alias_sample
7 7 from tqdm import tqdm
8 8 import argparse
9 9 import multiprocessing
... ...