GraphGym
GraphGym copied to clipboard
About Scenario 2
Thanks for the exciting program!
I am suffering from finding the optimal GNN model on the node classification task. This problem is caused by too much freedom of choice within and between layers. In other words, there are too many models to choose from and too many hyperparameters to optimize.
Referring to ogb's leaderboard to find the optimal model is a potential solution, but as the paper showed,
the best GNN designs for different tasks differ drastically.
From my understanding, GraphGym has provided the idea that similar tasks can share the optimal model design.
As mentioned,
GraphGym provides a simple interface to try out thousands of GNNs in parallel and understand the best designs for your specific task. GraphGym also recommends a "go-to" GNN design space, after investigating 10 million GNN model-task combinations.
I would like to know if there are off-the-shelf model-task combinations that I can use directly, without using the interface to try out GNN designs.
Thanks for the question. This is a good feature request. Currently, I recommend trying out the design space of 96 GNN models, shown here: https://github.com/snap-stanford/GraphGym/blob/master/run/grids/example.txt
You could take a look at the jupyter notebook here, regarding how to compute task similarity: https://github.com/snap-stanford/GraphGym/blob/master/analysis/design_space.ipynb
We are also actively working on enriching the functionalities of GraphGym.