HomeDocsAPI Reference
Kumo.ai
Docs

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_modeDefault Value
FAST[32]
NORMAL[32]
BEST[32]