finally fixed n2v bug

Author: VHRanger

csrgraph/graph.py

@@ -80,7 +80,7 @@ def __init__(self, data, nodenames=None, copy=True, threads=0):
             else:
                 self.names = nodenames
         # NetworkX Graph input
-        elif isinstance(data, nx.Graph):
+        elif isinstance(data, (nx.Graph, nx.DiGraph)):
             mat = nx.adj_matrix(data)
             mat.data = mat.data.astype(np.float32)
             self.mat = mat

csrgraph/random_walks.py

@@ -188,9 +188,10 @@ def _node2vec_walks(Tdata, Tindptr, Tindices,
                 return_weight, neighbor_weight
             )
         # Write final states
-        res[i, k] = state
+        res[i, -1] = state
     return res
 
+
 @jit(nopython=True)
 def graph_from_walks_inner(walks):
     """