Even smaller k-NN model training

Signed-off-by: Thomas Farr <[email protected]>
opensearch-project · Feb 27, 2025 · 065fe26 · 065fe26
1 parent 076e657
commit 065fe26
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/tests/default/knn/models.yaml b/tests/default/knn/models.yaml
@@ -27,10 +27,6 @@ prologues:
         - {recommendation_vector: [1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5], duration: 12.2}
         - {index: {_index: movies, _id: '2'}}
         - {recommendation_vector: [2.5, 2.5, 2.5, 2.5, 2.5, 2.5, 2.5, 2.5], duration: 7.1}
-        - {index: {_index: movies, _id: '3'}}
-        - {recommendation_vector: [3.5, 3.5, 3.5, 3.5, 3.5, 3.5, 3.5, 3.5], duration: 4.8}
-        - {index: {_index: movies, _id: '4'}}
-        - {recommendation_vector: [4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5], duration: 18.6}
     status: [200]
   - method: POST
     path: /_plugins/_knn/models/model-1/_train
@@ -49,11 +45,11 @@ prologues:
           name: ivf
           engine: faiss
           parameters:
-            nlist: 4
+            nlist: 2
             encoder:
               name: pq
               parameters:
-                code_size: 2
+                code_size: 1
     status: [200]
 epilogues:
   - path: /movies

diff --git a/tests/default/knn/models/search.yaml b/tests/default/knn/models/search.yaml
@@ -27,10 +27,6 @@ prologues:
         - {recommendation_vector: [1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5, 1.5], duration: 12.2}
         - {index: {_index: movies, _id: '2'}}
         - {recommendation_vector: [2.5, 2.5, 2.5, 2.5, 2.5, 2.5, 2.5, 2.5], duration: 7.1}
-        - {index: {_index: movies, _id: '3'}}
-        - {recommendation_vector: [3.5, 3.5, 3.5, 3.5, 3.5, 3.5, 3.5, 3.5], duration: 4.8}
-        - {index: {_index: movies, _id: '4'}}
-        - {recommendation_vector: [4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5, 4.5], duration: 18.6}
     status: [200]
   - method: POST
     path: /_plugins/_knn/models/model-1/_train
@@ -49,11 +45,11 @@ prologues:
           name: ivf
           engine: faiss
           parameters:
-            nlist: 4
+            nlist: 2
             encoder:
               name: pq
               parameters:
-                code_size: 2
+                code_size: 1
     status: [200]
 epilogues:
   - path: /movies