Add FLAT index

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 44f1a4fec3
1 changed files with 2 additions and 0 deletions
--- a/milvus_client.py
+++ b/milvus_client.py
@ -52,6 +52,7 @@ class MilvusClient(PyOperator):
            self.kwargs['limit'] = 10

        index_params = {
+            'FLAT': {'params': {'nprobe': 10}},
            'IVF_FLAT': {'params': {'nprobe': 10}},
            'IVF_SQ8': {'params': {'nprobe': 10}},
            'IVF_PQ': {'params': {'nprobe': 10}},
@ -76,6 +77,7 @@ class MilvusClient(PyOperator):
                self.kwargs['param']['metric_type'] = 'L2'

    def __call__(self, query: 'ndarray'):
+        self._collection.load()
        milvus_result = self._collection.search(
            data=[query],
            **self.kwargs