def __init__(self, full_graph, ntype, ctype, hidden_dims, n_layers, gat_num_heads): super().__init__() self.nodeproj = layers.LinearProjector(full_graph, ntype, hidden_dims) self.contextproj = layers.LinearProjector(full_graph, ctype, hidden_dims) self.multisage = layers.MultiSAGENet(hidden_dims, n_layers, gat_num_heads) self.scorer = layers.ItemToItemScorer(full_graph, ntype)
def __init__(self, full_graph, ntype, textsets, hidden_dims, n_layers): super().__init__() self.proj = layers.LinearProjector(full_graph, ntype, textsets, hidden_dims) self.sage = layers.SAGENet(hidden_dims, n_layers) self.scorer = layers.ItemToItemScorer(full_graph, ntype)