Elasticsearch 官方自带的向量搜索压测

配置情况

elasticsearch 版本7.14.1，16G JVM

创建 mapping

{
    "feature":{
        "path_match":"feature",
        "mapping":{
            "type":"dense_vector",
            "dims":256
        }
    }
}

query 语句

GET features_self_1w/_search
{
  "size": 1,
  "query": {
    "script_score": {
      "query": {
        "match_all": {}
      },
      "script": {
        "source": "doc['feature'].size() ==0 ? 0 : cosineSimilarity(params.query_vector, 'feature') + 1.0",
        "params": {
          "query_vector": [
            -0.291748046875,
            0.096435546875,
            -0.11865234375,
            0.107177734375,
            -0.117919921875,
            ...
            ...
            ...
            -0.105712890625,
            -0.01171875
          ]
        }
      }
    }
  }
}