mirror of
https://github.com/gnh1201/welsonjs.git
synced 2024-11-26 23:41:42 +00:00
51 lines
1.3 KiB
JavaScript
51 lines
1.3 KiB
JavaScript
|
// lib/extramath.js
|
||
|
// https://github.com/gnh1201/welsonjs
|
||
|
|
||
|
function DTM() {
|
||
|
this.data = [];
|
||
|
this.terms = [];
|
||
|
|
||
|
this.add = function(s) {
|
||
|
var w = s.trim().split(/\s+/);
|
||
|
for (var i = 0; i < w.length; i++) {
|
||
|
if (this.terms.indexOf(w[i]) < 0) this.terms.push(w[i]);
|
||
|
}
|
||
|
this.data.push(w);
|
||
|
};
|
||
|
|
||
|
this.toDTM = function() {
|
||
|
var dtm = [];
|
||
|
for (var i = 0; i < this.data.length; i++) {
|
||
|
var dt = [];
|
||
|
for (var k = 0; k < this.terms.length; k++) {
|
||
|
dt.push(this.data[i].indexOf(this.terms[k]) < 0 ? 0 : 1);
|
||
|
}
|
||
|
dtm.push(dt);
|
||
|
}
|
||
|
return dtm;
|
||
|
};
|
||
|
}
|
||
|
|
||
|
function cosineSim(A, B) {
|
||
|
var dotproduct = 0;
|
||
|
var mA = 0;
|
||
|
var mB = 0;
|
||
|
for (i = 0; i < A.length; i++) { // here you missed the i++
|
||
|
dotproduct += (A[i] * B[i]);
|
||
|
mA += (A[i] * A[i]);
|
||
|
mB += (B[i] * B[i]);
|
||
|
}
|
||
|
mA = Math.sqrt(mA);
|
||
|
mB = Math.sqrt(mB);
|
||
|
var similarity = (dotproduct) / ((mA)*(mB)) // here you needed extra brackets
|
||
|
return similarity;
|
||
|
}
|
||
|
|
||
|
exports.DTM = DTM;
|
||
|
exports.cosineSim = cosineSim;
|
||
|
|
||
|
exports.VERSIONINFO = "ExtraMath module (extramath.js) version 0.0.1";
|
||
|
exports.AUTHOR = "catswords@protonmail.com";
|
||
|
exports.global = global;
|
||
|
exports.require = global.require;
|