Skip to navigation

Can I use an external model to embed a column?

View as Markdown

Yes. Compute the embeddings before ingestion and store one fixed-length vector of floating-point values per row. Assign the column stype="sequence" and use the NumericalList encoder.

Use the same embedding model and preprocessing during training and prediction. See Sequence and Embedding Encoder for configuration.

To generate entity vectors from a trained Kumo model, see Embedding Outputs.