Datasets
GNNGraphs.jl doesn't come with its own datasets, but leverages those available in the Julia (and non-Julia) ecosystem. In particular, the examples in the GraphNeuralNetworks.jl repository make use of the MLDatasets.jl package. There you will find common graph datasets such as Cora, PubMed, Citeseer, TUDataset and many others. For graphs with static structures and temporal features, datasets such as METRLA, PEMSBAY, ChickenPox, and WindMillEnergy are available. For graphs featuring both temporal structures and temporal features, the TemporalBrains dataset is suitable.
GraphNeuralNetworks.jl provides the mldataset2gnngraph
method for interfacing with MLDatasets.jl.
GNNGraphs.mldataset2gnngraph
— Functionmldataset2gnngraph(dataset)
Convert a graph dataset from the package MLDatasets.jl into one or many GNNGraph
s.
Examples
julia> using MLDatasets, GraphNeuralNetworks
julia> mldataset2gnngraph(Cora())
GNNGraph:
num_nodes = 2708
num_edges = 10556
ndata:
features => 1433×2708 Matrix{Float32}
targets => 2708-element Vector{Int64}
train_mask => 2708-element BitVector
val_mask => 2708-element BitVector
test_mask => 2708-element BitVector