decision2vec-neg-signed-no-attrs-jm.py 文件源码

def read_graph():
    if args.directed:
        g = nx.read_edgelist(args.input, nodetype=str, data=[('Release_Detained', int)], create_using=nx.DiGraph())
    else:
        g = nx.read_edgelist(args.input, nodetype=str, data=[('Release_Detained', int)], create_using=nx.Graph())
    # instantiate MyGraph using G
    myg = MyGraph(g, args.dimensions, args.window_size, is_directed=False, node_attributes=False, walks_per_node=args.num_walks)
    print 'Graph created...'
    return myg