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)

相关推荐