test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-12-03 04:19:18 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

[skip ci] foramt pass

Browse Source
This commit is contained in:
yhz 2019-11-20 11:23:48 +08:00
parent a5a41212f8
commit ed5219b728
1 changed files with 31 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
core/src/sdk/examples/partition/src/ClientTest.cpp
View File

@ -31,45 +31,45 @@
namespace { namespace {
const char *TABLE_NAME = milvus_sdk::Utils::GenTableName().c_str(); const char* TABLE_NAME = milvus_sdk::Utils::GenTableName().c_str();
constexpr int64_t TABLE_DIMENSION = 512; constexpr int64_t TABLE_DIMENSION = 512;
constexpr int64_t TABLE_INDEX_FILE_SIZE = 1024; constexpr int64_t TABLE_INDEX_FILE_SIZE = 1024;
constexpr milvus::MetricType TABLE_METRIC_TYPE = milvus::MetricType::L2; constexpr milvus::MetricType TABLE_METRIC_TYPE = milvus::MetricType::L2;
constexpr int64_t BATCH_ROW_COUNT = 10000; constexpr int64_t BATCH_ROW_COUNT = 10000;
constexpr int64_t NQ = 5; constexpr int64_t NQ = 5;
constexpr int64_t TOP_K = 10; constexpr int64_t TOP_K = 10;
constexpr int64_t NPROBE = 32; constexpr int64_t NPROBE = 32;
constexpr int64_t SEARCH_TARGET = 5000; // change this value, result is different constexpr int64_t SEARCH_TARGET = 5000; // change this value, result is different
constexpr milvus::IndexType INDEX_TYPE = milvus::IndexType::IVFSQ8; constexpr milvus::IndexType INDEX_TYPE = milvus::IndexType::IVFSQ8;
constexpr int32_t N_LIST = 15000; constexpr int32_t N_LIST = 15000;
constexpr int32_t PARTITION_COUNT = 5; constexpr int32_t PARTITION_COUNT = 5;
constexpr int32_t TARGET_PARTITION = 3; constexpr int32_t TARGET_PARTITION = 3;
milvus::TableSchema milvus::TableSchema
BuildTableSchema() { BuildTableSchema() {
milvus::TableSchema tb_schema = {TABLE_NAME, TABLE_DIMENSION, TABLE_INDEX_FILE_SIZE, TABLE_METRIC_TYPE}; milvus::TableSchema tb_schema = {TABLE_NAME, TABLE_DIMENSION, TABLE_INDEX_FILE_SIZE, TABLE_METRIC_TYPE};
return tb_schema; return tb_schema;
} }
milvus::PartitionParam milvus::PartitionParam
BuildPartitionParam(int32_t index) { BuildPartitionParam(int32_t index) {
std::string tag = std::to_string(index); std::string tag = std::to_string(index);
std::string partition_name = std::string(TABLE_NAME) + "_" + tag; std::string partition_name = std::string(TABLE_NAME) + "_" + tag;
milvus::PartitionParam partition_param = {TABLE_NAME, partition_name, tag}; milvus::PartitionParam partition_param = {TABLE_NAME, partition_name, tag};
return partition_param; return partition_param;
} }
milvus::IndexParam milvus::IndexParam
BuildIndexParam() { BuildIndexParam() {
milvus::IndexParam index_param = {TABLE_NAME, INDEX_TYPE, N_LIST}; milvus::IndexParam index_param = {TABLE_NAME, INDEX_TYPE, N_LIST};
return index_param; return index_param;
} }
} // namespace } // namespace
void void
ClientTest::Test(const std::string &address, const std::string &port) { ClientTest::Test(const std::string& address, const std::string& port) {
std::shared_ptr<milvus::Connection> conn = milvus::Connection::Create(); std::shared_ptr<milvus::Connection> conn = milvus::Connection::Create();
milvus::Status stat; milvus::Status stat;