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