From c03b94a22e38936a3c0f0cb766f1e4b37c6a4c44 Mon Sep 17 00:00:00 2001 From: kenneth Date: Thu, 4 Apr 2024 08:37:54 +0200 Subject: [PATCH] use model dimension --- .../hsfs/knn_search/news-search-knn-save-model.ipynb | 4 ++-- api_examples/hsfs/knn_search/news-search-knn.ipynb | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/api_examples/hsfs/knn_search/news-search-knn-save-model.ipynb b/api_examples/hsfs/knn_search/news-search-knn-save-model.ipynb index 8a2c455d..d2a82644 100644 --- a/api_examples/hsfs/knn_search/news-search-knn-save-model.ipynb +++ b/api_examples/hsfs/knn_search/news-search-knn-save-model.ipynb @@ -197,8 +197,8 @@ "\n", "emb = embedding.EmbeddingIndex(index_name=f\"news_fg_{version}\")\n", "# specify the name, dimension, and model of the embedding features \n", - "emb.add_embedding(\"embedding_body\", len(df[\"embedding_body\"][0]), model=hsml_model)\n", - "emb.add_embedding(\"embedding_heading\", len(df[\"embedding_heading\"][0]), model=hsml_model)" + "emb.add_embedding(\"embedding_body\", model.get_sentence_embedding_dimension(), model=hsml_model)\n", + "emb.add_embedding(\"embedding_heading\", model.get_sentence_embedding_dimension(), model=hsml_model)" ] }, { diff --git a/api_examples/hsfs/knn_search/news-search-knn.ipynb b/api_examples/hsfs/knn_search/news-search-knn.ipynb index 89724613..988056c7 100644 --- a/api_examples/hsfs/knn_search/news-search-knn.ipynb +++ b/api_examples/hsfs/knn_search/news-search-knn.ipynb @@ -189,8 +189,8 @@ "outputs": [], "source": [ "# specify the name and dimension of the embedding features \n", - "emb.add_embedding(\"embedding_body\", len(df[\"embedding_body\"][0]))\n", - "emb.add_embedding(\"embedding_heading\", len(df[\"embedding_heading\"][0]))" + "emb.add_embedding(\"embedding_body\", model.get_sentence_embedding_dimension())\n", + "emb.add_embedding(\"embedding_heading\", model.get_sentence_embedding_dimension())" ] }, { @@ -340,7 +340,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.10.0" + "version": "3.10.13" } }, "nbformat": 4,