HomeDocsAPI Reference
Kumo.ai
Docs

target_embedding_mode

target_embedding_mode: (lookup (default) | feature | fusion) (Optional)

Description

Specifies the mode of target node embeddings to use in the final link prediction calculation (i.e., dot product). The mode should be either lookup, feature, or fusion.

  • By default, it is set to lookup mode, which will use lookup/shallow embeddings for target nodes.
  • The feature mode will generate target node embeddings based purely on target node features.
  • The fusion mode will fuse the the lookup embeddings and feature-based embeddings.

Supported Task Types

  • Link Prediction