How to train the model on new dataset?
Opened this issue · 2 comments
JanineCHEN commented
Hi, great work!
I am interested in testing the model on my own graph datasets with both node and edge features, could you kindly provide some instructions on how to accomplish this? Namely what data format criteria need to be satisfied and how to load customized datasets for training?
Any help would be greatly appreciated!
JanineCHEN commented
@xguo7 Hi, could you share some more details of the data format you used for training. Many thanks in advance!
xguo7 commented
@xguo7 Hi, could you share some more details of the data format you used for training. Many thanks in advance!
Hi Janine,
The input graph is represented by a node feature matrix and the edge feature tensor. You could check the dimension in the "problem formulation" part in our paper.