Function HeterGraphSageEncoder

src/model/encoder/graphsage_encoder.cc:116–132

Different namespaces use different encoders, and then concat the output of different encoders as the final representation

// Different namespaces use different encoders, and then concat the output of
// different encoders as the final representation
GraphNode* HeterGraphSageEncoder(const id_name_t& id_2_name,
                                 const std::string& prefix,
                                 const std::vector<GroupConfigItem3>& items,
                                 int depth, bool use_neigh_feat, bool sparse,
                                 double relu_alpha, int dim) {
  std::vector<GraphNode*> hiddens;
  GraphNode* hidden = nullptr;
  for (auto& entry : id_2_name) {
    auto ns_id = entry.first;
    auto ns_name = entry.second;
    hidden = GraphSageEncoder(prefix + std::to_string(ns_id) + ns_name, items,
                              depth, use_neigh_feat, sparse, relu_alpha, dim);
    hiddens.emplace_back(hidden);
  }
  auto* next_hidden = Concat("", hiddens, 0);
  return next_hidden;
}

}  // namespace embedx