Method __init__

tf_euler/python/utils/encoders.py:239–260

(self, metapath, dim, aggregator='attention',
                 feature_idx=-1, feature_dim=0, max_id=-1, use_id=False,
                 sparse_feature_idx=-1, sparse_feature_max_id=-1,
                 embedding_dim=16, use_hash_embedding=False,
                 use_residual=False, head_num=4,
                 **kwargs)

class GenieEncoder(GCNEncoder):
    def __init__(self, metapath, dim, aggregator='attention',
                 feature_idx=-1, feature_dim=0, max_id=-1, use_id=False,
                 sparse_feature_idx=-1, sparse_feature_max_id=-1,
                 embedding_dim=16, use_hash_embedding=False,
                 use_residual=False, head_num=4,
                 **kwargs):
        super(GenieEncoder, self).__init__(metapath, dim,
                                           aggregator,
                                           feature_idx,
                                           feature_dim,
                                           max_id,
                                           use_id,
                                           sparse_feature_idx,
                                           sparse_feature_max_id,
                                           embedding_dim,
                                           use_hash_embedding,
                                           use_residual,
                                           head_num, **kwargs)
        self.dim = dim
        self.depth_fc = []
        for layer in range(self.num_layers + 1):
            self.depth_fc.append(layers.Dense(dim))

    def call(self, inputs):
        nodes, adjs = euler_ops.get_multi_hop_neighbor(inputs, self.metapath)