From f30c3273866ed2d036f304b16dfac8c4543d953a Mon Sep 17 00:00:00 2001
From: yu yunfeng <yunfeng.yu@zilliz.com>
Date: Tue, 2 Jul 2019 15:03:21 +0800
Subject: [PATCH] Update faiss parameter

Former-commit-id: a1ae1ac1bc963c18ac8a5281f8aa07ecba50eb94
---
 cpp/src/wrapper/Index.cpp   | 4 ++++
 cpp/src/wrapper/Operand.cpp | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/cpp/src/wrapper/Index.cpp b/cpp/src/wrapper/Index.cpp
index 35dd646fdf..c306b65e1a 100644
--- a/cpp/src/wrapper/Index.cpp
+++ b/cpp/src/wrapper/Index.cpp
@@ -13,6 +13,7 @@
 
 #include "Index.h"
 #include "faiss/index_io.h"
+#include "faiss/IndexIVF.h"
 
 namespace zilliz {
 namespace milvus {
@@ -55,6 +56,9 @@ bool Index::add_with_ids(idx_t n, const float *xdata, const long *xids) {
 
 bool Index::search(idx_t n, const float *data, idx_t k, float *distances, long *labels) const {
     try {
+        if(auto ivf_index = std::dynamic_pointer_cast<faiss::IndexIVF>(index_)) {
+            ivf_index->nprobe = 100;
+        }
         index_->search(n, data, k, distances, labels);
     }
     catch (std::exception &e) {
diff --git a/cpp/src/wrapper/Operand.cpp b/cpp/src/wrapper/Operand.cpp
index 25341676a6..30e31067fd 100644
--- a/cpp/src/wrapper/Operand.cpp
+++ b/cpp/src/wrapper/Operand.cpp
@@ -39,7 +39,7 @@ string Operand::get_index_type(const int &nb) {
         }
         case IVF: {
             index_str += (ncent != 0 ? index_type + std::to_string(ncent) :
-                          index_type + std::to_string(int(nb / 1000000.0 * 16384)));
+                          index_type + std::to_string(int(nb / 1000000.0 * 1638)));
             break;
         }
         case IDMAP: {