gammagl.models.GraceModel

class GraceModel(in_feat, hid_feat, out_feat, num_layers, activation, temp)[source]
get_loss(z1, z2)[source]
sim(z1, z2)[source]
get_embeding(feat, edge, weight, num_nodes)[source]
forward(feat1, edge1, weight1, num_node1, feat2, edge2, weight2, num_node2)[source]