Skip to content

Node embeddings extraction #1

@DiegoOrtego

Description

@DiegoOrtego

Hi!

First of all, congrats for the great work here and for sharing the code. I'm playing with the code and found that that you are using "text-embedding-3-small" as embedding model for vector search, which is not a free model. You are providing embeddings for nodes, but not for the queries, so I cannot run the full code (maybe I miss something). Anyway, I want to have full control of the models being used so wanted to try out with Qwen3-Embeddings to replace "text-embedding-3-small". However, I realized that there is no code for embedding extraction for the nodes. Could you guide me in how to do it? Do I just need to embed the node text ("texts" field from 'node_embeddings_add_rel_not_compact.pt' and 'node_embeddings_no_rel_not_compact.pt')? I guess that "texts" field in 'node_embeddings_add_rel_not_compact.pt' already contains the relational information to expand the node information with 2-hop info as you describe in the paper, right?

Many thanks in advance!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions