coco
164 lines
10 KiB
JavaScript
164 lines
10 KiB
JavaScript
var SVC = function(nClasses, nRows, vectors, coefficients, intercepts, weights, kernel, gamma, coef0, degree) {
|
|
|
|
this.nClasses = nClasses;
|
|
this.classes = new Array(nClasses);
|
|
for (var i = 0; i < nClasses; i++) {
|
|
this.classes[i] = i;
|
|
}
|
|
this.nRows = nRows;
|
|
this.vectors = vectors;
|
|
this.coefficients = coefficients;
|
|
this.intercepts = intercepts;
|
|
this.weights = weights;
|
|
this.kernel = kernel.toUpperCase();
|
|
this.gamma = gamma;
|
|
this.coef0 = coef0;
|
|
this.degree = degree;
|
|
|
|
this.predict = function(features) {
|
|
|
|
var kernels = new Array(vectors.length);
|
|
var kernel;
|
|
switch (this.kernel) {
|
|
case 'LINEAR':
|
|
// <x,x'>
|
|
for (var i = 0; i < this.vectors.length; i++) {
|
|
kernel = 0.;
|
|
for (var j = 0; j < this.vectors[i].length; j++) {
|
|
kernel += this.vectors[i][j] * features[j];
|
|
}
|
|
kernels[i] = kernel;
|
|
}
|
|
break;
|
|
case 'POLY':
|
|
// (y<x,x'>+r)^d
|
|
for (var i = 0; i < this.vectors.length; i++) {
|
|
kernel = 0.;
|
|
for (var j = 0; j < this.vectors[i].length; j++) {
|
|
kernel += this.vectors[i][j] * features[j];
|
|
}
|
|
kernels[i] = Math.pow((this.gamma * kernel) + this.coef0, this.degree);
|
|
}
|
|
break;
|
|
case 'RBF':
|
|
// exp(-y|x-x'|^2)
|
|
for (var i = 0; i < this.vectors.length; i++) {
|
|
kernel = 0.;
|
|
for (var j = 0; j < this.vectors[i].length; j++) {
|
|
kernel += Math.pow(this.vectors[i][j] - features[j], 2);
|
|
}
|
|
kernels[i] = Math.exp(-this.gamma * kernel);
|
|
}
|
|
break;
|
|
case 'SIGMOID':
|
|
// tanh(y<x,x'>+r)
|
|
for (var i = 0; i < this.vectors.length; i++) {
|
|
kernel = 0.;
|
|
for (var j = 0; j < this.vectors[i].length; j++) {
|
|
kernel += this.vectors[i][j] * features[j];
|
|
}
|
|
kernels[i] = Math.tanh((this.gamma * kernel) + this.coef0);
|
|
}
|
|
break;
|
|
}
|
|
|
|
var starts = new Array(this.nRows);
|
|
for (var i = 0; i < this.nRows; i++) {
|
|
if (i != 0) {
|
|
var start = 0;
|
|
for (var j = 0; j < i; j++) {
|
|
start += this.weights[j];
|
|
}
|
|
starts[i] = start;
|
|
} else {
|
|
starts[0] = 0;
|
|
}
|
|
}
|
|
|
|
var ends = new Array(this.nRows);
|
|
for (var i = 0; i < this.nRows; i++) {
|
|
ends[i] = this.weights[i] + starts[i];
|
|
}
|
|
|
|
if (this.nClasses == 2) {
|
|
|
|
for (var i = 0; i < kernels.length; i++) {
|
|
kernels[i] = -kernels[i];
|
|
}
|
|
|
|
var decision = 0.;
|
|
for (var k = starts[1]; k < ends[1]; k++) {
|
|
decision += kernels[k] * this.coefficients[0][k];
|
|
}
|
|
for (var k = starts[0]; k < ends[0]; k++) {
|
|
decision += kernels[k] * this.coefficients[0][k];
|
|
}
|
|
decision += this.intercepts[0];
|
|
|
|
if (decision > 0) {
|
|
return 0;
|
|
}
|
|
return 1;
|
|
|
|
}
|
|
|
|
var decisions = new Array(this.intercepts.length);
|
|
for (var i = 0, d = 0, l = this.nRows; i < l; i++) {
|
|
for (var j = i + 1; j < l; j++) {
|
|
var tmp = 0.;
|
|
for (var k = starts[j]; k < ends[j]; k++) {
|
|
tmp += this.coefficients[i][k] * kernels[k];
|
|
}
|
|
for (var k = starts[i]; k < ends[i]; k++) {
|
|
tmp += this.coefficients[j - 1][k] * kernels[k];
|
|
}
|
|
decisions[d] = tmp + this.intercepts[d];
|
|
d++;
|
|
}
|
|
}
|
|
|
|
var votes = new Array(this.intercepts.length);
|
|
for (var i = 0, d = 0, l = this.nRows; i < l; i++) {
|
|
for (var j = i + 1; j < l; j++) {
|
|
votes[d] = decisions[d] > 0 ? i : j;
|
|
d++;
|
|
}
|
|
}
|
|
|
|
var amounts = new Array(this.nClasses).fill(0);
|
|
for (var i = 0, l = votes.length; i < l; i++) {
|
|
amounts[votes[i]] += 1;
|
|
}
|
|
|
|
var classVal = -1, classIdx = -1;
|
|
for (var i = 0, l = amounts.length; i < l; i++) {
|
|
if (amounts[i] > classVal) {
|
|
classVal = amounts[i];
|
|
classIdx= i;
|
|
}
|
|
}
|
|
return this.classes[classIdx];
|
|
|
|
}
|
|
|
|
};
|
|
|
|
if (typeof process !== 'undefined' && typeof process.argv !== 'undefined') {
|
|
if (process.argv.length - 2 === 6) {
|
|
|
|
// Features:
|
|
var features = process.argv.slice(2);
|
|
|
|
// Parameters:
|
|
var vectors = [[16.0, 170.0, 213.0, 170.0, 144.0, 178.0], [14.0, 204.0, 179.0, 130.0, 153.0, 177.0], [14.0, 202.0, 181.0, 132.0, 153.0, 176.0], [17.0, 168.0, 218.0, 175.0, 143.0, 178.0], [17.0, 193.0, 250.0, 194.0, 147.0, 193.0], [15.0, 203.0, 182.0, 135.0, 150.0, 178.0], [15.0, 212.0, 183.0, 134.0, 152.0, 180.0], [18.0, 211.0, 165.0, 131.0, 142.0, 179.0], [17.0, 207.0, 183.0, 142.0, 145.0, 181.0], [19.0, 232.0, 179.0, 142.0, 142.0, 186.0], [20.0, 235.0, 186.0, 147.0, 142.0, 189.0], [20.0, 191.0, 213.0, 177.0, 136.0, 188.0], [18.0, 216.0, 216.0, 166.0, 147.0, 191.0], [23.0, 186.0, 165.0, 148.0, 128.0, 179.0], [23.0, 169.0, 166.0, 152.0, 127.0, 175.0], [21.0, 188.0, 191.0, 165.0, 132.0, 183.0], [20.0, 203.0, 161.0, 135.0, 135.0, 178.0], [21.0, 227.0, 204.0, 168.0, 136.0, 194.0], [23.0, 215.0, 165.0, 144.0, 130.0, 184.0], [28.0, 175.0, 112.0, 112.0, 120.0, 167.0], [26.0, 212.0, 166.0, 156.0, 122.0, 188.0], [26.0, 171.0, 141.0, 137.0, 121.0, 173.0], [28.0, 163.0, 123.0, 124.0, 120.0, 168.0], [28.0, 166.0, 122.0, 123.0, 119.0, 168.0], [42.0, 132.0, 94.0, 95.0, 110.0, 153.0], [43.0, 137.0, 92.0, 93.0, 110.0, 153.0], [34.0, 166.0, 128.0, 130.0, 111.0, 170.0], [42.0, 133.0, 99.0, 100.0, 109.0, 154.0], [37.0, 54.0, 140.0, 146.0, 119.0, 142.0], [36.0, 150.0, 143.0, 145.0, 109.0, 170.0], [34.0, 161.0, 119.0, 122.0, 113.0, 167.0], [47.0, 106.0, 112.0, 112.0, 108.0, 151.0], [48.0, 112.0, 120.0, 119.0, 105.0, 153.0], [67.0, 117.0, 80.0, 78.0, 107.0, 140.0], [49.0, 122.0, 110.0, 109.0, 104.0, 153.0], [48.0, 95.0, 133.0, 133.0, 107.0, 151.0], [61.0, 80.0, 104.0, 104.0, 108.0, 142.0], [63.0, 67.0, 129.0, 129.0, 109.0, 140.0], [65.0, 103.0, 96.0, 95.0, 106.0, 142.0], [73.0, 149.0, 62.0, 59.0, 109.0, 136.0], [74.0, 146.0, 87.0, 84.0, 104.0, 137.0], [61.0, 73.0, 123.0, 123.0, 108.0, 142.0]];
|
|
var coefficients = [[0.0, 0.0, 0.0, 0.0, 0.007123731733845719, 0.0, 0.026728670619480987, -0.0, -0.022642062294742756, -0.0, -0.0, -0.0, -0.011210340058583986, -0.0, -0.0, -0.004492810017798028, -0.0, -0.0, -0.0, -0.0, -0.0009822727926227658, -0.00019697324708030917, -0.0, -0.0, -0.0, -0.0, -0.00024315366079156689, -0.0, -0.0, -7.887982104716923e-05, -0.0, -0.0, -0.0001269056378111678, -0.0, -0.0, -0.0, -0.0, -2.224093446634136e-05, -0.0, -0.0, -7.305730597946722e-05, -0.0], [0.0, 0.0, 0.0, 0.0023541594215438543, 0.0, 0.0021386505962541743, 0.0, 0.0, 0.0, 0.0, 0.04303144632801865, 0.021990515109160533, 0.0, -0.0, -0.0, -0.0, -0.017016118801059996, -0.04800584263611915, -0.0, -0.0, -0.002683681315761449, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0004261666311527815, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.00013782299883790497, -0.0, -0.0, -1.246285746727607e-05, -0.0, -0.0, -9.438411501101229e-05, -0.0], [0.0, 0.0, 0.0, 3.689360823356543e-05, 0.0, 0.0011423524314695091, 0.0, 0.001430025142535212, 0.0, 0.0005948943046227278, 0.0, 0.000658761868603509, 0.0, 0.008324876468227304, 0.0, 0.0, 0.0, 0.0, 0.007713657530346414, -0.0, -0.011605206592790481, -0.004433327405783237, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0013726835964310295, -0.0, -0.0, -0.0002149116477739606, -0.0, -0.0, -3.402872408940634e-05, -0.0, -7.993845202285296e-05, -0.0, -0.0, -7.21000410116088e-05, -0.0], [0.0, 0.0, 0.0, 0.0, 0.0, 0.0003220334818387361, 0.0, 0.0004261666311527815, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0013726835964310295, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.009096123618122409, 0.012708095634696546, -0.0, -0.0, -0.006079604041865676, -0.0, -0.0, -0.0, -0.015724615210953283, -0.0, -0.0, -0.0, -0.0006673340006673339, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0003358099684919548, -3.0082337769457922e-05], [3.13346695773095e-05, 0.0, 9.557096823385828e-05, 0.0, 0.0, 0.0, 0.0, 0.00013782299883790497, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.00024894037186336693, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0006673340006673339, 0.0, 0.005440788430323922, 0.00020463528718228497, 0.0, 0.013800715626807585, 0.00911414574180425, 0.0, 0.0, -0.028560285086118048, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0, -0.0030207789307972795], [0.0, 9.529824044580858e-05, 0.0, 0.0, 0.0, 0.0, 0.0, 0.00010684697247828835, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.00015203849303446176, 0.0, 0.0, 0.0, 0.0, 0.0003658923062614127, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0009514754328023643, 0.0, 0.0, 0.002069303497994916, 0.0, 0.0, 0.0, 0.0, 0.3587936835959387, 0.0, 0.024271536378842457, -0.12581162030930793, -0.0, -0.08810534750995824, -0.1691482521555165, -0.0, -0.0]];
|
|
var intercepts = [-6.002192891642206, -3.8628284042206995, -3.333075888643083, -6.954002229759305, -4.416097880179401, -3.6114319594536997, -21.442641474459478, -4.668420984774443, -8.14383123801408, -4.5443688291828215, -3.8049761670220867, -8.128105076816803, -14.037748798901934, -6.013597288024034, -4.323534693677444, -22.78835291091018, -8.33867200533867, -5.1963460012753275, -8.616928192852473, -5.692778997252465, 37.37043070272911];
|
|
var weights = [7, 6, 6, 5, 7, 5, 6];
|
|
|
|
// Prediction:
|
|
var clf = new SVC(7, 7, vectors, coefficients, intercepts, weights, "linear", 0.02, 0.0, 3);
|
|
var prediction = clf.predict(features);
|
|
console.log(prediction);
|
|
|
|
}
|
|
} |