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;

}