Kumo.ai home page
Search...
⌘K
Ask AI
Community
Contact Support
Start for free
Start for free
Search...
Navigation
Model Architecture
output_embedding_dim
Guide
Examples
Reference
FAQ
Releases
Predictive Query
Overview
Primary Commands
Aggregation Operators
Boolean Operators
Model Plan
Column Processing
Model Architecture
GNN
Graph Transformers
Selecting Link Prediction Model Architectures on Kumo
activation
distance_measure
handle_new_target_entities
module
normalization
num_post_message_passing_layers
num_pre_message_passing_layers
ranking_embedding_loss_coeff
output_embedding_dim
prediction_time_encodings
target_embedding_mode
use_seq_id
Neighbor Sampling
Optimization
Training Job Plan
Training Table Generation
SDK
SDK Reference
On this page
output_embedding_dim: <list[int]> (Optional)
Description
Supported Task Types
Default Values
Model Architecture
output_embedding_dim
output_embedding_dim: <list[int]>
(Optional)
Description
A list of output embedding dimensions for link prediction models for AutoML to explore.
output_embedding_dim
must satisfy
0 < output_embedding <= 256
.
Supported Task Types
Link Prediction
Default Values
run_mode
Default Value
FAST
[32]
NORMAL
[32]
BEST
[32]
ranking_embedding_loss_coeff
prediction_time_encodings
Assistant
Responses are generated using AI and may contain mistakes.