Cugraph

wget https://snap.stanford.edu/data/cit-Patents.txt.gz

gunzip cit-Patents.txt.gz

 

import pandas as pd

pandas_edgelist = pd.read_csv(

    "cit-Patents.txt",

    skiprows=4,

    delimiter="\t",

    names=["src", "dst"],

    dtype={"src": "int32", "dst": "int32"},

)


import cugraph as cg

G = cg.from_pandas_edgelist(

    pandas_edgelist, source="src", destination="dst", create_using=cg.Graph(directed=True)

)


import time

st = time.time()

bc_result = cg.betweenness_centrality(G, k=1)

print(f"{(time.time() - st):.6f} s")