gammagl.models.GraphSAGE_Sample_Model

class GraphSAGE_Sample_Model(in_feat, hid_feat, out_feat, drop_rate, num_layers, name=None)

The GraphSAGE operator from the “Inductive Representation Learning on Large Graphs” paper

Parameters:

• in_feat – number of input feature.

• hid_feat – number of hidden feature.

• out_feat – number of output feature.

• drop_rate – dropout rate.

• num_layers – number of sage layers.

• name – model name.

forward(x, edgeIndices)

inference(feat, dataloader, cur_x)