Dataset endpoints

class graphdatascience.procedure_surface.api.catalog.dataset_endpoints.DatasetEndpoints
load_cora(graph_name: str = 'cora', undirected: bool = False) → GraphV2

A citation network introduced.

Parameters:

  • graph_name (str) – Name of the graph to be created

  • undirected (bool) – Whether the graph should be undirected

Returns:

A handle to the graph.

Return type:

GraphV2

load_imdb(graph_name: str = 'imdb', undirected: bool = True) → GraphV2

A heterogeneous graph that is used to benchmark node classification or link prediction models.

The graph contains Actors, Directors, Movies (and UnclassifiedMovies) as nodes, and relationships between actors and movies that they acted in, and between directors and movies which they directed for.

Parameters:

  • graph_name (str) – Name of the graph to be created

  • undirected (bool) – Whether the graph should be undirected

Returns:

A handle to the graph.

Return type:

GraphV2

load_karate_club(graph_name: str = 'karate_club', undirected: bool = False) → GraphV2

A social network introduced by http://konect.cc/networks/ucidata-zachary/[Zachary].

Parameters:

  • graph_name (str) – Name of the graph to be created

  • undirected (bool) – Whether the graph should be undirected

Returns:

A handle to the graph.

Return type:

GraphV2

load_lastfm(graph_name: str = 'lastfm', undirected: bool = True) → GraphV2

A heterogeneous graph that is used to benchmark link prediction models. The original raw data is from http://www.lastfm.com/[LastFM].

The graph contains User and Artists as nodes, each with a rawId field that corresponds to the ids from the raw data. The rawId could be used, for example, to check the artist names by referring back to the HetRec’11 .dat files.

Parameters:

  • graph_name (str) – Name of the graph to be created

  • undirected (bool) – Whether the graph should be undirected

Returns:

A handle to the graph.

Return type:

GraphV2

property networkx

Convenience wrapper to load networkx graphs into the graph catalog.

Return type:

NXLoader

property ogbl: OGBLLoader

Datasets used for link property prediction.

property ogbn: OGBNLoader

Datasets used for node property prediction.