DBLP8k

class dhg.data.DBLP8k(data_root=None)[source]

Bases: dhg.data.base.BaseData

The DBLP-8k dataset is a citation network dataset for link prediction task. The dataset is a part of the dataset crawled according to DBLP API, and we have selected each item based on some conditions, such as the venue and publication year (from 2018 to 2022). It contains 6498 authors and 2603 papers.

The content of the DBLP-8k dataset includes the following:

num_vertices: The number of vertices: \(8,657\).
num_edges: The number of edges: \(2,603\).
edge_list: The edge list. List with length \(2,603\).

Parameters: data_root (str, optional) – The data_root has stored the data. If set to None, this function will auto-download from server and save into the default direction ~/.dhg/datasets/. Defaults to None.