Skip to main contenttarget_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
Default Values
| run_mode | Default Value |
|---|
| FAST | [lookup] |
| NORMAL | [lookup] |
| BEST | [lookup] |