I want to run MariusGNN to train a GraphSAGE model with neighbor sampling dropout. As far as I understand, I need to define in the yaml config file GNN layers which describe number of neighbors to sample (fanouts) and dropout rate. However, I find that the max_neighbor option is only available in UNIFORM type of NeighborSamplingLayer, and rate option is only available in DROPOUT type. Is there a way to define both options for a GraphSAGE layer?
I want to run MariusGNN to train a GraphSAGE model with neighbor sampling dropout. As far as I understand, I need to define in the yaml config file GNN layers which describe number of neighbors to sample (fanouts) and dropout rate. However, I find that the
max_neighboroption is only available inUNIFORMtype ofNeighborSamplingLayer, andrateoption is only available inDROPOUTtype. Is there a way to define both options for a GraphSAGE layer?