@antv/algorithm/lib/cosine-similarity.js

graph algorithm

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = void 0;
var _vector = _interopRequireDefault(require("./utils/vector"));
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
/**
 * cosine-similarity算法 计算余弦相似度
 * @param item 元素
 * @param targetItem 目标元素
 */
var cosineSimilarity = function cosineSimilarity(item, targetItem) {
  // 目标元素向量
  var targetItemVector = new _vector.default(targetItem);
  // 目标元素向量的模长
  var targetNodeNorm2 = targetItemVector.norm2();
  // 元素向量
  var itemVector = new _vector.default(item);
  // 元素向量的模长
  var itemNorm2 = itemVector.norm2();
  // 计算元素向量和目标元素向量的点积
  var dot = targetItemVector.dot(itemVector);
  var norm2Product = targetNodeNorm2 * itemNorm2;
  // 计算元素向量和目标元素向量的余弦相似度
  var cosineSimilarity = norm2Product ? dot / norm2Product : 0;
  return cosineSimilarity;
};
var _default = cosineSimilarity;
exports.default = _default;