Skip to content

Open Graph Benchmark integration #162

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
DoktorMike opened this issue Apr 23, 2022 · 2 comments · Fixed by #164
Closed

Open Graph Benchmark integration #162

DoktorMike opened this issue Apr 23, 2022 · 2 comments · Fixed by #164

Comments

@DoktorMike
Copy link

There are quite a few useful datasets, benchmarks and leaderboards in ogb. The paper is here.

From what I can see it plays nicely with PyTorch Geometric and Deep Graph Library both of which are Python packages.

I would think that having access to these resources through GraphNeuralNetworks.jl or another julia package could ease the attraction of new users. I haven't used many of these datasets myself so I don't know more than this.

@CarloLucibello
Copy link
Member

They are already available in MLDatasets.jl. It should be quite easy to create a custom function converting MLDatasets' type to GNN.jl's types.

Once JuliaML/MLDatasets.jl#114 is merged, the interface of MLDatasets' datasets will be streamlined and I will be able to implement conversion utilities here without having to depend on MLDatasets directly.

@CarloLucibello
Copy link
Member

With #164 we have

julia> using MLDatasets, GraphNeuralNetworks

julia> dataset = OGBDataset("ogbn-arxiv")
dataset OGBDataset:
  name        =>    ogbn-arxiv
  metadata    =>    Dict{String, Any} with 17 entries
  graphs      =>    1-element Vector{MLDatasets.Graph}
  graph_data  =>    nothing

julia> mldataset2gnngraph(dataset)
GNNGraph:
    num_nodes = 169343
    num_edges = 1166243
    ndata:
        val_mask => 169343-element BitVector
        test_mask => 169343-element BitVector
        year => 169343-element Vector{Int64}
        features => 128×169343 Matrix{Float32}
        label => 169343-element Vector{Int64}
        train_mask => 169343-element BitVector

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants