开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > .net编程 > 查看源码
svm-toy.cpp
资源名称：svm.rar　[点击查看]
上传用户：xgw_05
上传日期：2014-12-08
资源大小：2726k
文件大小：10k
源码类别：

.net编程

开发平台：

Java

svm-toy.cpp：源码内容
  1. #include <qapplication.h>
  2. #include <qpainter.h>
  3. #include <qpixmap.h>
  4. #include <qwidget.h>
  5. #include <qpushbutton.h>
  6. #include <qlineedit.h>
  7. #include <qfiledialog.h>
  8. #include <stdio.h>
  9. #include <stdlib.h>
  10. #include <ctype.h>
  11. #include <list>
  12. #include "../../svm.h"
  13. #define DEFAULT_PARAM "-t 2 -c 100"
  14. #define XLEN 500
  15. #define YLEN 500
  17. QRgb colors[] =
  18. {
  19. qRgb(0,0,0),
  20. qRgb(0,120,120),
  21. qRgb(120,120,0),
  22. qRgb(120,0,120),
  23. qRgb(0,200,200),
  24. qRgb(200,200,0),
  25. qRgb(200,0,200)
  26. };
  28. class SvmToyWindow : public QWidget
  29. {
  31. Q_OBJECT
  33. public:
  34. SvmToyWindow();
  35. ~SvmToyWindow();
  36. protected:
  37. virtual void mousePressEvent( QMouseEvent* );
  38. virtual void paintEvent( QPaintEvent* );
  40. private:
  41. QPixmap buffer;
  42. QPixmap icon1;
  43. QPixmap icon2;
  44. QPixmap icon3;
  45. QPushButton button_change_icon;
  46. QPushButton button_run;
  47. QPushButton button_clear;
  48. QPushButton button_save;
  49. QPushButton button_load;
  50. QLineEdit input_line;
  51. QPainter window_painter;
  52. QPainter buffer_painter;
  53. struct point {
  54. double x, y;
  55. signed char value;
  56. };
  57. list<point> point_list;
  58. int current_value;
  59. const QPixmap& choose_icon(int v)
  60. {
  61. if(v==1) return icon1;
  62. else if(v==2) return icon2;
  63. else return icon3;
  64. }
  65. void clear_all()
  66. {
  67. point_list.clear();
  68. buffer.fill(black);
  69. paintEvent(NULL);
  70. }
  71. void draw_point(const point& p)
  72. {
  73. const QPixmap& icon = choose_icon(p.value);
  74. window_painter.drawPixmap((int)(p.x*XLEN),(int)(p.y*YLEN),icon);
  75. buffer_painter.drawPixmap((int)(p.x*XLEN),(int)(p.y*YLEN),icon);
  76. }
  77. void draw_all_points()
  78. {
  79. for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++)
  80. draw_point(*p);
  81. }
  82. private slots: 
  83. void button_change_icon_clicked()
  84. {
  85. ++current_value;
  86. if(current_value > 3) current_value = 1;
  87. button_change_icon.setPixmap(choose_icon(current_value));
  88. }
  89. void button_run_clicked()
  90. {
  91. // guard
  92. if(point_list.empty()) return;
  94. svm_parameter param;
  95. int i,j;
  97. // default values
  98. param.svm_type = C_SVC;
  99. param.kernel_type = RBF;
  100. param.degree = 3;
  101. param.gamma = 0;
  102. param.coef0 = 0;
  103. param.nu = 0.5;
  104. param.cache_size = 40;
  105. param.C = 1;
  106. param.eps = 1e-3;
  107. param.p = 0.1;
  108. param.shrinking = 1;
  109. param.nr_weight = 0;
  110. param.weight_label = NULL;
  111. param.weight = NULL;
  113. // parse options
  114. const char *p = input_line.text();
  116. while (1) {
  117. while (*p && *p != '-')
  118. p++;
  120. if (*p == '')
  121. break;
  123. p++;
  124. switch (*p++) {
  125. case 's':
  126. param.svm_type = atoi(p);
  127. break;
  128. case 't':
  129. param.kernel_type = atoi(p);
  130. break;
  131. case 'd':
  132. param.degree = atof(p);
  133. break;
  134. case 'g':
  135. param.gamma = atof(p);
  136. break;
  137. case 'r':
  138. param.coef0 = atof(p);
  139. break;
  140. case 'n':
  141. param.nu = atof(p);
  142. break;
  143. case 'm':
  144. param.cache_size = atof(p);
  145. break;
  146. case 'c':
  147. param.C = atof(p);
  148. break;
  149. case 'e':
  150. param.eps = atof(p);
  151. break;
  152. case 'p':
  153. param.p = atof(p);
  154. break;
  155. case 'h':
  156. param.shrinking = atoi(p);
  157. break;
  158. case 'w':
  159. ++param.nr_weight;
  160. param.weight_label = (int *)realloc(param.weight_label,sizeof(int)*param.nr_weight);
  161. param.weight = (double *)realloc(param.weight,sizeof(double)*param.nr_weight);
  162. param.weight_label[param.nr_weight-1] = atoi(p);
  163. while(*p && !isspace(*p)) ++p;
  164. param.weight[param.nr_weight-1] = atof(p);
  165. break;
  166. }
  167. }
  169. // build problem
  170. svm_problem prob;
  172. prob.l = point_list.size();
  173. prob.y = new double[prob.l];
  175. if(param.svm_type == EPSILON_SVR ||
  176.    param.svm_type == NU_SVR)
  177. {
  178. if(param.gamma == 0) param.gamma = 1;
  179. svm_node *x_space = new svm_node[2 * prob.l];
  180. prob.x = new svm_node *[prob.l];
  182. i = 0;
  183. for (list <point>::iterator q = point_list.begin(); q != point_list.end(); q++, i++)
  184. {
  185. x_space[2 * i].index = 1;
  186. x_space[2 * i].value = q->x;
  187. x_space[2 * i + 1].index = -1;
  188. prob.x[i] = &x_space[2 * i];
  189. prob.y[i] = q->y;
  190. }
  192. // build model & classify
  193. svm_model *model = svm_train(&prob, &param);
  194. svm_node x[2];
  195. x[0].index = 1;
  196. x[1].index = -1;
  197. int *j = new int[XLEN];
  199. for (i = 0; i < XLEN; i++)
  200. {
  201. x[0].value = (double) i / XLEN;
  202. j[i] = (int)(YLEN*svm_predict(model, x));
  203. }
  205. buffer_painter.setPen(colors[0]);
  206. buffer_painter.drawLine(0,0,0,YLEN-1);
  207. window_painter.setPen(colors[0]);
  208. window_painter.drawLine(0,0,0,YLEN-1);
  210. int p = (int)(param.p * YLEN);
  211. for(i = 1; i < XLEN; i++)
  212. {
  213. buffer_painter.setPen(colors[0]);
  214. buffer_painter.drawLine(i,0,i,YLEN-1);
  215. window_painter.setPen(colors[0]);
  216. window_painter.drawLine(i,0,i,YLEN-1);
  218. buffer_painter.setPen(colors[5]);
  219. buffer_painter.drawLine(i-1,j[i-1],i,j[i]);
  220. window_painter.setPen(colors[5]);
  221. window_painter.drawLine(i-1,j[i-1],i,j[i]);
  223. if(param.svm_type == EPSILON_SVR)
  224. {
  225. buffer_painter.setPen(colors[2]);
  226. buffer_painter.drawLine(i-1,j[i-1]+p,i,j[i]+p);
  227. window_painter.setPen(colors[2]);
  228. window_painter.drawLine(i-1,j[i-1]+p,i,j[i]+p);
  230. buffer_painter.setPen(colors[2]);
  231. buffer_painter.drawLine(i-1,j[i-1]-p,i,j[i]-p);
  232. window_painter.setPen(colors[2]);
  233. window_painter.drawLine(i-1,j[i-1]-p,i,j[i]-p);
  234. }
  235. }
  237. svm_destroy_model(model);
  238. delete[] j;
  239. delete[] x_space;
  240. delete[] prob.x;
  241. delete[] prob.y;
  242. }
  243. else
  244. {
  245. if(param.gamma == 0) param.gamma = 0.5;
  246. svm_node *x_space = new svm_node[3 * prob.l];
  247. prob.x = new svm_node *[prob.l];
  249. i = 0;
  250. for (list <point>::iterator q = point_list.begin(); q != point_list.end(); q++, i++)
  251. {
  252. x_space[3 * i].index = 1;
  253. x_space[3 * i].value = q->x;
  254. x_space[3 * i + 1].index = 2;
  255. x_space[3 * i + 1].value = q->y;
  256. x_space[3 * i + 2].index = -1;
  257. prob.x[i] = &x_space[3 * i];
  258. prob.y[i] = q->value;
  259. }
  261. // build model & classify
  262. svm_model *model = svm_train(&prob, &param);
  263. svm_node x[3];
  264. x[0].index = 1;
  265. x[1].index = 2;
  266. x[2].index = -1;
  268. for (i = 0; i < XLEN; i++)
  269. for (j = 0; j < YLEN ; j++) {
  270. x[0].value = (double) i / XLEN;
  271. x[1].value = (double) j / YLEN;
  272. double d = svm_predict(model, x);
  273. buffer_painter.setPen(colors[(int)d]);
  274. window_painter.setPen(colors[(int)d]);
  275. buffer_painter.drawPoint(i,j);
  276. window_painter.drawPoint(i,j);
  277. }
  279. svm_destroy_model(model);
  280. delete[] x_space;
  281. delete[] prob.x;
  282. delete[] prob.y;
  283. }
  284. free(param.weight_label);
  285. free(param.weight);
  286. draw_all_points();
  287. }
  288. void button_clear_clicked()
  289. {
  290. clear_all();
  291. }
  292. void button_save_clicked()
  293. {
  294. QString filename = QFileDialog::getSaveFileName();
  295. if(!filename.isNull())
  296. {
  297. FILE *fp = fopen(filename,"w");
  298. if(fp)
  299. {
  300. for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++)
  301. fprintf(fp,"%d 1:%f 2:%fn", p->value, p->x, p->y);
  302. fclose(fp);
  303. }
  304. }
  305. }
  306. void button_load_clicked()
  307. {
  308. QString filename = QFileDialog::getOpenFileName();
  309. if(!filename.isNull())
  310. {
  311. FILE *fp = fopen(filename,"r");
  312. if(fp)
  313. {
  314. clear_all();
  315. char buf[4096];
  316. while(fgets(buf,sizeof(buf),fp))
  317. {
  318. int v;
  319. double x,y;
  320. if(sscanf(buf,"%d%*d:%lf%*d:%lf",&v,&x,&y)!=3)
  321. break;
  322. point p = {x,y,v};
  323. point_list.push_back(p);
  324. }
  325. fclose(fp);
  326. draw_all_points();
  327. }
  328. }
  330. }
  331. };
  333. #include "svm-toy.moc"
  335. SvmToyWindow::SvmToyWindow()
  336. :button_change_icon(this)
  337. ,button_run("Run",this)
  338. ,button_clear("Clear",this)
  339. ,button_save("Save",this)
  340. ,button_load("Load",this)
  341. ,input_line(this)
  342. ,current_value(1)
  343. {
  344. buffer.resize(XLEN,YLEN);
  345. buffer.fill(black);
  347. window_painter.begin(this);
  348. buffer_painter.begin(&buffer);
  350. QObject::connect(&button_change_icon, SIGNAL(clicked()), this,
  351.  SLOT(button_change_icon_clicked()));
  352. QObject::connect(&button_run, SIGNAL(clicked()), this,
  353.  SLOT(button_run_clicked()));
  354. QObject::connect(&button_clear, SIGNAL(clicked()), this,
  355.  SLOT(button_clear_clicked()));
  356. QObject::connect(&button_save, SIGNAL(clicked()), this,
  357.  SLOT(button_save_clicked()));
  358. QObject::connect(&button_load, SIGNAL(clicked()), this,
  359.  SLOT(button_load_clicked()));
  360. QObject::connect(&input_line, SIGNAL(returnPressed()), this,
  361.  SLOT(button_run_clicked()));
  363.    // don't blank the window before repainting
  364. setBackgroundMode( NoBackground );
  365.   
  366. icon1.resize(4,4);
  367. icon2.resize(4,4);
  368. icon3.resize(4,4);
  370. QPainter painter;
  371. painter.begin(&icon1);
  372. painter.fillRect(0,0,4,4,QBrush(colors[4]));
  373. painter.end();
  375. painter.begin(&icon2);
  376. painter.fillRect(0,0,4,4,QBrush(colors[5]));
  377. painter.end();
  379. painter.begin(&icon3);
  380. painter.fillRect(0,0,4,4,QBrush(colors[6]));
  381. painter.end();
  383. button_change_icon.setGeometry( 0, YLEN, 50, 25 );
  384. button_run.setGeometry( 50, YLEN, 50, 25 );
  385. button_clear.setGeometry( 100, YLEN, 50, 25 );
  386. button_save.setGeometry( 150, YLEN, 50, 25);
  387. button_load.setGeometry( 200, YLEN, 50, 25);
  388. input_line.setGeometry( 250, YLEN, 250, 25);
  390. input_line.setText(DEFAULT_PARAM);
  391. button_change_icon.setPixmap(icon1);
  392. }
  394. SvmToyWindow::~SvmToyWindow()
  395. {
  396. window_painter.end();
  397. buffer_painter.end();
  398. }
  400. void SvmToyWindow::mousePressEvent( QMouseEvent* event )
  401. {
  402. point p = {(double)event->x()/XLEN, (double)event->y()/YLEN, current_value};
  403. point_list.push_back(p);
  404. draw_point(p);
  405. }
  407. void SvmToyWindow::paintEvent( QPaintEvent* )
  408. {
  409. // copy the image from the buffer pixmap to the window
  410. bitBlt( this, 0, 0, &buffer );
  411. }
  413. int main( int argc, char* argv[] )
  414. {
  415. QApplication myapp( argc, argv );
  417. SvmToyWindow* mywidget = new SvmToyWindow();
  418. mywidget->setGeometry( 100, 100, XLEN, YLEN+25 );
  420. myapp.setMainWidget( mywidget );
  421. mywidget->show();
  422. return myapp.exec();
  423. }