diff --git a/sdk/examples/simple/insert.cpp b/sdk/examples/simple/insert.cpp index b19b70513b0ee5e695eed4ef97251fed5ad4d183..86c1967d28fc718d10eeeef5b98c8b0b7c0c2c5d 100644 --- a/sdk/examples/simple/insert.cpp +++ b/sdk/examples/simple/insert.cpp @@ -19,6 +19,7 @@ #include "grpc/ClientProxy.h" #include "interface/ConnectionImpl.h" #include "utils/TimeRecorder.h" +#include const int N = 200000; const int DIM = 16; @@ -31,9 +32,14 @@ const milvus::FieldValue GetData() { for (int i = 0; i < N; i++) { int32_data.push_back(i); } + std::default_random_engine eng(rand() % 20); + std::normal_distribution dis(0, 1); std::vector vector_data; for (int i = 0; i < N; i++) { - std::vector float_data(DIM, 10.25); + std::vector float_data(DIM); + for(auto &x: float_data) { + x = dis(eng); + } milvus::VectorData vectorData; vectorData.float_data = float_data; vector_data.push_back(vectorData); diff --git a/sdk/examples/simple/search.cpp b/sdk/examples/simple/search.cpp index f15781fa288835374506b91c09e8ff3c423ee201..1da315232abe8525c9817d9848128e9f50a481b1 100644 --- a/sdk/examples/simple/search.cpp +++ b/sdk/examples/simple/search.cpp @@ -17,6 +17,7 @@ #include "interface/ConnectionImpl.h" #include "utils/TimeRecorder.h" #include "utils/Utils.h" +#include const int TOP_K = 10; @@ -38,16 +39,19 @@ int main(int argc , char**argv) { partition_list.emplace_back("partition-3"); milvus::VectorParam vectorParam; - milvus::VectorData vectorData; - std::vector float_data; - for (int i = 0; i < 100; ++i) { - float_data.emplace_back(i); - } - - vectorData.float_data = float_data; - std::vector vector_records; + + std::default_random_engine eng(rand() % 20); + std::normal_distribution dis(0, 1); + for (int j = 0; j < 10; ++j) { + milvus::VectorData vectorData; + std::vector float_data; + for (int i = 0; i < 100; ++i) { + float_data.emplace_back(dis(eng)); + } + + vectorData.float_data = float_data; vector_records.emplace_back(vectorData); }