Function EgesEncoder

src/model/encoder/eges_encoder.cc:17–27

namespace embedx {

GraphNode* EgesEncoder(const std::string& prefix, GraphNode* Xsrc_feat,
                       GraphNode* Xsrc_node,
                       const std::vector<GroupConfigItem3>& items,
                       bool sparse) {
  auto* feat_embed = XInputGroupEmbeddingLookup(prefix + "node_feat", Xsrc_feat,
                                                items, sparse);
  auto* group_weight = XNodeGroupWeightLookup(prefix + "group_weights",
                                              Xsrc_node, items, sparse);
  auto* src_embed = WeightedAverage("", feat_embed, group_weight);
  return src_embed;
}

}  // namespace embedx