Skip to main content

attach_embeddings

Function attach_embeddings 

Source
pub fn attach_embeddings(
    batch: &RecordBatch,
    embeddings: &[(String, Vec<f32>)],
) -> Result<RecordBatch>
Expand description

Update a CodeNodes RecordBatch with embedding vectors.

Replaces the null embedding column with actual vectors for nodes that have embeddings. Nodes without embeddings remain null.