bknyaz / graph_nn

Graph Classification with Graph Convolutional Networks in PyTorch (NeurIPS 2018 Workshop)

Home Page:https://arxiv.org/abs/1811.09595

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is all datasets in /data are supported ?

zhaohan-xi opened this issue · comments

Hi Dear Author,

This looks a very good implementation in using GCN for graph-level classification (may also called inductive). Besides the sample running you shown in jupyter notebook, are the other datasets are also supported in running? If yes, could I ask which part should modify to change the calling of datasets? Thanks!

Hi,
The DataReader class can in principle read any dataset in the TU format (see datasets here https://ls11-www.cs.tu-dortmund.de/staff/morris/graphkerneldatasets).
You need to download the dataset, put it in the ./data/ folder and change line dataset = 'proteins'.
But I think it's better to use the python script in my repo (https://github.com/bknyaz/graph_nn/blob/master/graph_unet.py), because I tested it more and for different datasets. It has the --dataset argument that you can use to specify any TU dataset.

Got, thanks for response