huihun/graphsage-tf
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #11 from bkj/patch-1

Browse Source 
Add networkx version checking to utils.py
This commit is contained in:
William L Hamilton 2017-11-02 14:41:17 -07:00 committed by GitHub
commit bb355bb696
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
graphsage/utils.py
View File

@ -6,7 +6,10 @@ import json
import sys import sys
import os import os
import networkx as nx
from networkx.readwrite import json_graph from networkx.readwrite import json_graph
major, minor = map(int, nx.__version__.split('.'))
assert (major <= 1) and (minor <= 11), "networkx major version > 1.11"
WALK_LEN=5 WALK_LEN=5
N_WALKS=50 N_WALKS=50