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)
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)
