跳至內容

Selecting embedding size in BedrockEmbedding titan v1 and v2, switching dimensions dynamically via keyword arguments using langchain

0

I am trying to provide dimension argument while trying to fetch the Titan version 1 and version 2 embedding models, but the BedrockEmbedding class does not have the feature of dimension, It only provideds the default 1024 dimension. How can we solve this?

Below is how i am calling BedrockEmbedding from langchain_community.embeddings import BedrockEmbeddings

已提問 2 年前檢視次數 877 次

1 個回答
1

Hi,

You don't have to provide the embeddings dimensions in the query: the model that you select (via its id) will return the number of dimensions that it is programmed for.

So, if you work with multiple embedding engines, your code must be ready to work with variable numbers of dimensions depending on the model that you query.

Best,

Didier

專家

已回答 2 年前

專家

已審閱 2 年前

專家

已審閱 2 年前

  • Currently, our solution selects the model_id and defaults the text embedding dimensions to 1536 for Titan version 1 and 1024 for Titan version 2. However, we aim to test our dataset with various embedding sizes, such as 256, 512, and 1024.

    How can we dynamically set the embedding dimension for both Titan version 1 and version 2 using the Langchain Community BedrockEmbedding? The model keyword arguments do not seem to accept dimension specifications.

    How do we dynamically select the embedding dimension for both Titan version 1 and version 2 during testing.

您尚未登入。 登入 去張貼答案。

一個好的回答可以清楚地回答問題並提供建設性的意見回饋,同時有助於提問者的專業成長。