Implement graph regression using GIN code

Questions
#1

Hi,
I want to use GIN implementation of DGL for graph regression. I know I should change loss function to MSE loss and output_dim ==1.

my questions are

  1. should I add one more layer at the end of the GIN structure with output dimension =1?
  2. how should I change the gindt.py code specifically for this line self.gclasses = len(self.glabel_dict)
    I’m not sure how to use a regression target for every graph and glabel_dict at the same time.

Thanks,

2 Likes
#2

Hi @F_T ,
did you find any answer for your question I am trying to do the same thing using GIN
Thanks in advance