Groovy调用Weka生成决策树

Java代码

1. import weka.classifiers.Classifier
2. import weka.classifiers.trees.J48
3. import weka.core.Instances
4. import weka.core.converters.ArffLoader
5. import weka.gui.treevisualizer.PlaceNode2
6. import weka.gui.treevisualizer.TreeVisualizer
8. import javax.swing.JFrame
9. import java.awt.BorderLayout
11. /**
12. Weka的决策树分析功能测试
13. @author bruce.lin.chn
14. @version 2013-01-29
15. */
18. Classifier classifier= new J48()
19. ArffLoader arffLoader=new ArffLoader()
20. arffLoader.setFile(new File("../cpu.with.vendor.arff"))
22. Instances instancesTraining=arffLoader.getDataSet()
23. instancesTraining.setClassIndex(0)
25. println "共加载:"+instancesTraining.numInstances().toString()
26. classifier.buildClassifier(instancesTraining)
28. println "训练完毕!"
30. TreeVisualizer treeVisualizer=new TreeVisualizer(null,classifier.graph(),new PlaceNode2())
31. JFrame jFrame=new JFrame("决策树测试:J48")
32. jFrame.setDefaultCloSEOperation(JFrame.DISPOSE_ON_CLOSE)
33. jFrame.setSize(1024,600)
34. jFrame.getContentPane().setLayout(new BorderLayout())
35. jFrame.getContentPane().add(treeVisualizer,BorderLayout.CENTER)
37. jFrame.setVisible(true)
38. treeVisualizer.fitToScreen()

import weka.classifiers.Classifier
import weka.classifiers.trees.J48
import weka.core.Instances
import weka.core.converters.ArffLoader
import weka.gui.treevisualizer.PlaceNode2
import weka.gui.treevisualizer.TreeVisualizer

import javax.swing.JFrame
import java.awt.BorderLayout

/**
 Weka的决策树分析功能测试
 @author bruce.lin.chn
 @version 2013-01-29
*/


Classifier classifier= new J48()
ArffLoader arffLoader=new ArffLoader()
arffLoader.setFile(new File("../cpu.with.vendor.arff"))

Instances instancesTraining=arffLoader.getDataSet()
instancesTraining.setClassIndex(0)

println "共加载:"+instancesTraining.numInstances().toString()
classifier.buildClassifier(instancesTraining)

println "训练完毕!"

TreeVisualizer treeVisualizer=new TreeVisualizer(null,new PlaceNode2())
JFrame jFrame=new JFrame("决策树测试:J48")
jFrame.setDefaultCloSEOperation(JFrame.DISPOSE_ON_CLOSE)
jFrame.setSize(1024,600)
jFrame.getContentPane().setLayout(new BorderLayout())
jFrame.getContentPane().add(treeVisualizer,BorderLayout.CENTER)

jFrame.setVisible(true)
treeVisualizer.fitToScreen()