main.cpp

来自「基于SMO算法的支持向量机」· C++ 代码 · 共 23 行

#include "smo.h"

/*************************************************
 * 主函数，用来初始化变量，选择外层的权值用来循环
 ************************************************/
#define TRAIN //定义该参数表明是用来训练模型，否则用于预测
main(int argc,char* argv[])
{
#ifdef TRAIN
	smo("svm.data","svm.model");
#else
	load_svm("svm.model");
	vector<double> x;
	x.resize(2);
	x[0]=-11;
	x[1]=0;
	double y;
	y=predict_func(x);
	cout<<y<<endl;
#endif
}