Datasets

GNNGraphs.mldataset2gnngraph — Function

mldataset2gnngraph(dataset)

Convert a graph dataset from the package MLDatasets.jl into one or many GNNGraphs.

Examples

julia> using MLDatasets, GNNGraphs

julia> mldataset2gnngraph(Cora())
GNNGraph:
  num_nodes: 2708
  num_edges: 10556
  ndata:
        val_mask = 2708-element BitVector
        targets = 2708-element Vector{Int64}
        test_mask = 2708-element BitVector
        features = 1433×2708 Matrix{Float32}
        train_mask = 2708-element BitVector

source