ML之SVM:利用Js语言设计SVM算法(SMO算法+线性核/高斯核)
ML之SVM:利用Js语言设计SVM算法(SMO算法+线性核/高斯核)
输出结果
设计思路
设计代码(部分代码)
var doTest = function() {
loadData(); //调用loadData函数加载样本数据、初始化参数
svm.train(2000);//调用svm.train函数,迭代2000次实现SMO算法
var result = {};
//for循环实现得到并返回result
for(var x = -230; x <= 230; x += 5) {
for(var y = -230; y <= 230; y += 5) {
var sample = [x/100, y/100];
var pred = svm.pred(sample, 2);
if ( pred >= 1) {
pred = 2;
} else if ( pred <= -1) {
pred = -2;
} else if (pred > 0) {
pred = 1;
} else {
pred = -1;
}
result[x/100 + ',' + y/100] = pred;
}
}
return result;
}
赞 (0)