You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
1.1 KiB
31 lines
1.1 KiB
4 months ago
|
"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;
|