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

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

.net编程

开发平台：

Java

svm_toy.java：源码内容
  1. import libsvm.*;
  2. import java.applet.*;
  3. import java.awt.*;
  4. import java.util.*;
  5. import java.awt.event.*;
  6. import java.io.*;
  8. public class svm_toy extends Applet {
  10. static final String DEFAULT_PARAM="-t 2 -c 100";
  11. int XLEN;
  12. int YLEN;
  14. // off-screen buffer
  16. Image buffer;
  17. Graphics buffer_gc;
  19. // pre-allocated colors
  21. final static Color colors[] =
  22. {
  23.   new Color(0,0,0),
  24.   new Color(0,120,120),
  25.   new Color(120,120,0),
  26.   new Color(120,0,120),
  27.   new Color(0,200,200),
  28.   new Color(200,200,0),
  29.   new Color(200,0,200)
  30. };
  32. class point {
  33. point(double x, double y, byte value)
  34. {
  35. this.x = x;
  36. this.y = y;
  37. this.value = value;
  38. }
  39. double x, y;
  40. byte value;
  41. }
  43. Vector point_list = new Vector();
  44. byte current_value = 1;
  46. public void init()
  47. {
  48. Button button_change = new Button("Change");
  49. Button button_run = new Button("Run");
  50. Button button_clear = new Button("Clear");
  51. Button button_save = new Button("Save");
  52. Button button_load = new Button("Load");
  53. final TextField input_line = new TextField(DEFAULT_PARAM);
  55. BorderLayout layout = new BorderLayout();
  56. this.setLayout(layout);
  58. Panel p = new Panel();
  59. GridBagLayout gridbag = new GridBagLayout();
  60. p.setLayout(gridbag);
  62. GridBagConstraints c = new GridBagConstraints();
  63. c.fill = GridBagConstraints.HORIZONTAL;
  64. c.weightx = 1;
  65. c.gridwidth = 1;
  66. gridbag.setConstraints(button_change,c);
  67. gridbag.setConstraints(button_run,c);
  68. gridbag.setConstraints(button_clear,c);
  69. gridbag.setConstraints(button_save,c);
  70. gridbag.setConstraints(button_load,c);
  71. c.weightx = 5;
  72. c.gridwidth = 5;
  73. gridbag.setConstraints(input_line,c);
  75. p.add(button_change);
  76. p.add(button_run);
  77. p.add(button_clear);
  78. p.add(button_save);
  79. p.add(button_load);
  80. p.add(input_line);
  81. this.add(p,BorderLayout.SOUTH);
  83. button_change.addActionListener(new ActionListener()
  84. { public void actionPerformed (ActionEvent e)
  85.   { button_change_clicked(); }});
  87. button_run.addActionListener(new ActionListener()
  88. { public void actionPerformed (ActionEvent e)
  89.   { button_run_clicked(input_line.getText()); }});
  91. button_clear.addActionListener(new ActionListener()
  92. { public void actionPerformed (ActionEvent e)
  93.   { button_clear_clicked(); }});
  95. button_save.addActionListener(new ActionListener()
  96. { public void actionPerformed (ActionEvent e)
  97.   { button_save_clicked(); }});
  99. button_load.addActionListener(new ActionListener()
  100. { public void actionPerformed (ActionEvent e)
  101.   { button_load_clicked(); }});
  103. input_line.addActionListener(new ActionListener()
  104. { public void actionPerformed (ActionEvent e)
  105.   { button_run_clicked(input_line.getText()); }});
  107. this.enableEvents(AWTEvent.MOUSE_EVENT_MASK);
  108. }
  110. void draw_point(point p)
  111. {
  112. Color c = colors[p.value+3];
  114. Graphics window_gc = getGraphics();
  115. buffer_gc.setColor(c);
  116. buffer_gc.fillRect((int)(p.x*XLEN),(int)(p.y*YLEN),4,4);
  117. window_gc.setColor(c);
  118. window_gc.fillRect((int)(p.x*XLEN),(int)(p.y*YLEN),4,4);
  119. }
  121. void clear_all()
  122. {
  123. point_list.removeAllElements();
  124. if(buffer != null)
  125. {
  126. buffer_gc.setColor(colors[0]);
  127. buffer_gc.fillRect(0,0,XLEN,YLEN);
  128. }
  129. repaint();
  130. }
  132. void draw_all_points()
  133. {
  134. int n = point_list.size();
  135. for(int i=0;i<n;i++)
  136. draw_point((point)point_list.elementAt(i));
  137. }
  139. void button_change_clicked()
  140. {
  141. ++current_value;
  142. if(current_value > 3) current_value = 1;
  143. }
  145. private static double atof(String s)
  146. {
  147. return Double.valueOf(s).doubleValue();
  148. }
  150. private static int atoi(String s)
  151. {
  152. return Integer.parseInt(s);
  153. }
  155. void button_run_clicked(String args)
  156. {
  157. // guard
  158. if(point_list.isEmpty()) return;
  160. svm_parameter param = new svm_parameter();
  162. // default values
  163. param.svm_type = svm_parameter.C_SVC;
  164. param.kernel_type = svm_parameter.RBF;
  165. param.degree = 3;
  166. param.gamma = 0;
  167. param.coef0 = 0;
  168. param.nu = 0.5;
  169. param.cache_size = 40;
  170. param.C = 1;
  171. param.eps = 1e-3;
  172. param.p = 0.1;
  173. param.shrinking = 1;
  174. param.nr_weight = 0;
  175. param.weight_label = new int[0];
  176. param.weight = new double[0];
  178. // parse options
  179. StringTokenizer st = new StringTokenizer(args);
  180. String[] argv = new String[st.countTokens()];
  181. for(int i=0;i<argv.length;i++)
  182. argv[i] = st.nextToken();
  184. for(int i=0;i<argv.length;i++)
  185. {
  186. if(argv[i].charAt(0) != '-') break;
  187. ++i;
  188. switch(argv[i-1].charAt(1))
  189. {
  190. case 's':
  191. param.svm_type = atoi(argv[i]);
  192. break;
  193. case 't':
  194. param.kernel_type = atoi(argv[i]);
  195. break;
  196. case 'd':
  197. param.degree = atof(argv[i]);
  198. break;
  199. case 'g':
  200. param.gamma = atof(argv[i]);
  201. break;
  202. case 'r':
  203. param.coef0 = atof(argv[i]);
  204. break;
  205. case 'n':
  206. param.nu = atof(argv[i]);
  207. break;
  208. case 'm':
  209. param.cache_size = atof(argv[i]);
  210. break;
  211. case 'c':
  212. param.C = atof(argv[i]);
  213. break;
  214. case 'e':
  215. param.eps = atof(argv[i]);
  216. break;
  217. case 'p':
  218. param.p = atof(argv[i]);
  219. break;
  220. case 'h':
  221. param.shrinking = atoi(argv[i]);
  222. break;
  223. case 'w':
  224. ++param.nr_weight;
  225. {
  226. int[] old = param.weight_label;
  227. param.weight_label = new int[param.nr_weight];
  228. System.arraycopy(old,0,param.weight_label,0,param.nr_weight-1);
  229. }
  231. {
  232. double[] old = param.weight;
  233. param.weight = new double[param.nr_weight];
  234. System.arraycopy(old,0,param.weight,0,param.nr_weight-1);
  235. }
  237. param.weight_label[param.nr_weight-1] = atoi(argv[i-1].substring(2));
  238. param.weight[param.nr_weight-1] = atof(argv[i]);
  239. break;
  240. default:
  241. System.err.print("unknown optionn");
  242. }
  243. }
  245. // build problem
  246. svm_problem prob = new svm_problem();
  247. prob.l = point_list.size();
  248. prob.y = new double[prob.l];
  250. if(param.svm_type == svm_parameter.EPSILON_SVR ||
  251.    param.svm_type == svm_parameter.NU_SVR)
  252. {
  253. if(param.gamma == 0) param.gamma = 1;
  254. prob.x = new svm_node[prob.l][1];
  255. for(int i=0;i<prob.l;i++)
  256. {
  257. point p = (point)point_list.elementAt(i);
  258. prob.x[i][0] = new svm_node();
  259. prob.x[i][0].index = 1;
  260. prob.x[i][0].value = p.x;
  261. prob.y[i] = p.y;
  262. }
  264. // build model & classify
  265. svm_model model = svm.svm_train(prob, param);
  266. svm_node[] x = new svm_node[1];
  267. x[0] = new svm_node();
  268. x[0].index = 1;
  269. int[] j = new int[XLEN];
  271. Graphics window_gc = getGraphics();
  272. for (int i = 0; i < XLEN; i++)
  273. {
  274. x[0].value = (double) i / XLEN;
  275. j[i] = (int)(YLEN*svm.svm_predict(model, x));
  276. }
  278. buffer_gc.setColor(colors[0]);
  279. buffer_gc.drawLine(0,0,0,YLEN-1);
  280. window_gc.setColor(colors[0]);
  281. window_gc.drawLine(0,0,0,YLEN-1);
  283. int p = (int)(param.p * YLEN);
  284. for(int i=1;i<XLEN;i++)
  285. {
  286. buffer_gc.setColor(colors[0]);
  287. buffer_gc.drawLine(i,0,i,YLEN-1);
  288. window_gc.setColor(colors[0]);
  289. window_gc.drawLine(i,0,i,YLEN-1);
  291. buffer_gc.setColor(colors[5]);
  292. window_gc.setColor(colors[5]);
  293. buffer_gc.drawLine(i-1,j[i-1],i,j[i]);
  294. window_gc.drawLine(i-1,j[i-1],i,j[i]);
  296. if(param.svm_type == svm_parameter.EPSILON_SVR)
  297. {
  298. buffer_gc.setColor(colors[2]);
  299. window_gc.setColor(colors[2]);
  300. buffer_gc.drawLine(i-1,j[i-1]+p,i,j[i]+p);
  301. window_gc.drawLine(i-1,j[i-1]+p,i,j[i]+p);
  303. buffer_gc.setColor(colors[2]);
  304. window_gc.setColor(colors[2]);
  305. buffer_gc.drawLine(i-1,j[i-1]-p,i,j[i]-p);
  306. window_gc.drawLine(i-1,j[i-1]-p,i,j[i]-p);
  307. }
  308. }
  309. }
  310. else
  311. {
  312. if(param.gamma == 0) param.gamma = 0.5;
  313. prob.x = new svm_node [prob.l][2];
  314. for(int i=0;i<prob.l;i++)
  315. {
  316. point p = (point)point_list.elementAt(i);
  317. prob.x[i][0] = new svm_node();
  318. prob.x[i][0].index = 1;
  319. prob.x[i][0].value = p.x;
  320. prob.x[i][1] = new svm_node();
  321. prob.x[i][1].index = 2;
  322. prob.x[i][1].value = p.y;
  323. prob.y[i] = p.value;
  324. }
  326. // build model & classify
  327. svm_model model = svm.svm_train(prob, param);
  328. svm_node[] x = new svm_node[2];
  329. x[0] = new svm_node();
  330. x[1] = new svm_node();
  331. x[0].index = 1;
  332. x[1].index = 2;
  334. Graphics window_gc = getGraphics();
  335. for (int i = 0; i < XLEN; i++)
  336. for (int j = 0; j < YLEN ; j++) {
  337. x[0].value = (double) i / XLEN;
  338. x[1].value = (double) j / YLEN;
  339. double d = svm.svm_predict(model, x);
  340. buffer_gc.setColor(colors[(int)d]);
  341. window_gc.setColor(colors[(int)d]);
  342. buffer_gc.drawLine(i,j,i,j);
  343. window_gc.drawLine(i,j,i,j);
  344. }
  345. }
  347. draw_all_points();
  348. }
  350. void button_clear_clicked()
  351. {
  352. clear_all();
  353. }
  355. void button_save_clicked()
  356. {
  357. FileDialog dialog = new FileDialog(new Frame(),"Save",FileDialog.SAVE);
  358. dialog.setVisible(true);
  359. String filename = dialog.getFile();
  360. if (filename == null) return;
  361. try {
  362. DataOutputStream fp = new DataOutputStream(new FileOutputStream(filename));
  363. int n = point_list.size();
  364. for(int i=0;i<n;i++)
  365. {
  366. point p = (point)point_list.elementAt(i);
  367. fp.writeBytes(p.value+" 1:"+p.x+" 2:"+p.y+"n");
  368. }
  369. fp.close();
  370. } catch (IOException e) { System.err.print(e); }
  371. }
  373. void button_load_clicked()
  374. {
  375. FileDialog dialog = new FileDialog(new Frame(),"Load",FileDialog.LOAD);
  376. dialog.setVisible(true);
  377. String filename = dialog.getFile();
  378. if (filename == null) return;
  379. clear_all();
  380. try {
  381. BufferedReader fp = new BufferedReader(new FileReader(filename));
  382. String line;
  383. while((line = fp.readLine()) != null)
  384. {
  385. StringTokenizer st = new StringTokenizer(line," tnrf:");
  386. byte value = (byte)atoi(st.nextToken());
  387. st.nextToken();
  388. double x = atof(st.nextToken());
  389. st.nextToken();
  390. double y = atof(st.nextToken());
  391. point_list.addElement(new point(x,y,value));
  392. }
  393. fp.close();
  394. } catch (IOException e) { System.err.print(e); }
  395. draw_all_points();
  396. }
  398. protected void processMouseEvent(MouseEvent e)
  399. {
  400. if(e.getID() == MouseEvent.MOUSE_PRESSED)
  401. {
  402. if(e.getX() >= XLEN || e.getY() >= YLEN) return;
  403. point p = new point((double)e.getX()/XLEN,
  404.     (double)e.getY()/YLEN,
  405.     current_value);
  406. point_list.addElement(p);
  407. draw_point(p);
  408. }
  409. }
  411. public void paint(Graphics g)
  412. {
  413. // create buffer first time
  414. if(buffer == null) {
  415. buffer = this.createImage(XLEN,YLEN);
  416. buffer_gc = buffer.getGraphics();
  417. buffer_gc.setColor(colors[0]);
  418. buffer_gc.fillRect(0,0,XLEN,YLEN);
  419. }
  420. g.drawImage(buffer,0,0,this);
  421. }
  423. public Dimension getPreferredSize() { return new Dimension(XLEN,YLEN+50); }
  425. public void resize(Dimension d) { resize(d.width,d.height); }
  426. public void resize(int w,int h) {
  427. super.resize(w,h);
  428. XLEN = w;
  429. YLEN = h-50;
  430. clear_all();
  431. }
  433. public static void main(String[] argv)
  434. {
  435. new AppletFrame("svm_toy",new svm_toy(),500,500+50);
  436. }
  437. }
  439. class AppletFrame extends Frame {
  440. AppletFrame(String title, Applet applet, int width, int height)
  441. {
  442. super(title);
  443. applet.init();
  444. applet.resize(width,height);
  445. applet.start();
  446. this.add(applet);
  447. this.pack();
  448. this.show();
  449. }
  450. }