Elasticsearch - 基于浮点数组的余弦相似度排序

Question

是否可以根据两个不同浮点数组的余弦相似度进行排序？类似于如何通过传递坐标来按地理距离进行排序？

Answer 1

如果一个数组是输入，我是可能的，但您必须将余弦相似度实现为脚本：

  "script": {
    "lang": "painless",
    "source": """
      def vector = params._source[params.vector_field];
      def dot_product = 0.0;
      def v_norm = 0.0;
      for (int i = 0; i < params.query_vector.length; ++i) { 
          def x = vector[i]; 
          dot_product += x * params.query_vector[i]; 
          v_norm += x * x;
      }
      return v_norm > 0 ? dot_product / (params.query_v_norm * Math.sqrt(v_norm)) : -1;
"""
  }

但是，这使用了source 字段，这可能会很慢。 See this other question to make it faster