From 23c261ba576d9d9e8bab3b268aa73c3be7d0fc03 Mon Sep 17 00:00:00 2001 From: staoxiao <2906698981@qq.com> Date: Wed, 18 Oct 2023 02:58:29 +0800 Subject: [PATCH] Update bge_huggingface.ipynb (#8960) - Description: Considering the similarity computation method of [BGE](https://github.com/FlagOpen/FlagEmbedding) model is cosine similarity, set normalize_embeddings to be True. - Tag maintainer: @baskaryan Co-authored-by: Erick Friis --- docs/docs/integrations/text_embedding/bge_huggingface.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/docs/integrations/text_embedding/bge_huggingface.ipynb b/docs/docs/integrations/text_embedding/bge_huggingface.ipynb index 923ba92874..e42f11df9e 100644 --- a/docs/docs/integrations/text_embedding/bge_huggingface.ipynb +++ b/docs/docs/integrations/text_embedding/bge_huggingface.ipynb @@ -36,7 +36,7 @@ "\n", "model_name = \"BAAI/bge-small-en\"\n", "model_kwargs = {'device': 'cpu'}\n", - "encode_kwargs = {'normalize_embeddings': False}\n", + "encode_kwargs = {'normalize_embeddings': True}\n", "hf = HuggingFaceBgeEmbeddings(\n", " model_name=model_name,\n", " model_kwargs=model_kwargs,\n",