Commit 74330086a5206e5fca8fea386b1604a6c540281e
1 parent
e89d7a84
deepwalk refactor for memsave and perfermance optimize
Showing
2 changed files
with
9 additions
and
1 deletions
Show diff stats
offline_tasks/deepwalk/deepwalk.py
| @@ -3,7 +3,7 @@ import numpy as np | @@ -3,7 +3,7 @@ import numpy as np | ||
| 3 | import networkx as nx | 3 | import networkx as nx |
| 4 | from joblib import Parallel, delayed | 4 | from joblib import Parallel, delayed |
| 5 | import itertools | 5 | import itertools |
| 6 | -from alias import create_alias_table, alias_sample | 6 | +from .alias import create_alias_table, alias_sample |
| 7 | from tqdm import tqdm | 7 | from tqdm import tqdm |
| 8 | import argparse | 8 | import argparse |
| 9 | import multiprocessing | 9 | import multiprocessing |