In [1]:

    

%load_ext autoreload


    

In [2]:

    

#coding=utf-8
%autoreload

import numpy as np
from tqdm import tqdm
import pickle
from gensim.models import Word2Vec
import node2vec_layout
import networkx as nx

%matplotlib inline
import matplotlib.pyplot as plt
import matplotlib.cm as cm


    

In [3]:

    

adj_list = pickle.load(open('adj_list.pkl', 'rb'))
directed = True


    

In [5]:

    

G=nx.DiGraph()
list_nodes = []
node_index_in_G = {}
for v in adj_list:
    G.add_node(v)
    node_index_in_G[v] = len(node_index_in_G)
    list_nodes.append(v)
    for u in adj_list[v]:
        G.add_edge(v, u, weight = 1)


    

In [6]:

    

poses = node2vec_layout.node2vec_layout(adj_list)


    

In [8]:

    

plt.figure(figsize = (7, 7))
nx.draw(G, poses, node_size=1, width=0.1, with_labels=False, edge_color='gray');


    

In [ ]: