HomeDocsAPI Reference
Kumo.ai
Docs

target_embedding_mode

target_embedding_mode: list[lookup | feature | fusion] (Optional)

Description

A list of target embedding modes use in the final link prediction calculation, i.e., dot product, for AutoML to explore. The mode should be either lookup, feature, or fusion.

  • The lookup mode uses lookup embeddings for target nodes.
  • The feature mode generates target node embeddings based purely on target node features.
  • The fusion mode fuses the the lookup embeddings and feature-based embeddings.

Supported Task Types

  • Link Prediction

Default Values

run_modeDefault Value
FAST[lookup]
NORMAL[lookup]
BEST[lookup]