def from_numpy_array(A, create_using=Graph):
"""
Initializes the graph from numpy array containing adjacency matrix.
Set create_using to cugraph.DiGraph for directed graph and
cugraph.Graph for undirected Graph.
"""
if create_using is Graph:
G = Graph()
elif create_using is DiGraph:
G = DiGraph()
else:
raise Exception("create_using supports Graph and DiGraph")
G.from_numpy_array(A)
return G
def from_numpy_array(A, create_using=Graph):
"""
Initializes the graph from numpy array containing adjacency matrix.
Parameters
----------
A : numpy.array
A Numpy array that contains adjacency information
create_using: cugraph.DiGraph or cugraph.Graph, optional (default=Graph)
Indicate whether to create a directed or undirected graph
"""
if create_using is Graph:
G = Graph()
elif create_using is DiGraph:
G = DiGraph()
else:
raise Exception("create_using supports Graph and DiGraph")
G.from_numpy_array(A)
return G