Finition weka

src/main/Weka.java

@@ -1,7 +1,14 @@
 package main;
 
+import weka.classifiers.trees.J48;
 import weka.core.*;
+import weka.gui.treevisualizer.PlaceNode2;
+import weka.gui.treevisualizer.TreeVisualizer;
 
+import javax.swing.*;
+import java.awt.*;
+import java.awt.event.WindowAdapter;
+import java.awt.event.WindowEvent;
 import java.util.ArrayList;
 
 public class Weka {
@@ -66,5 +73,18 @@ public class Weka {
            System.out.println(String.valueOf(j.isChoisi()));
            isTrainingSet.add(i);
         }
+
+        J48 tree = new J48();
+        try {
+            tree.buildClassifier(isTrainingSet);
+            System.out.println(tree);
+            JFrame jf = new JFrame("Arbre");
+            jf.setSize(800, 800);
+            TreeVisualizer tv = new TreeVisualizer(null, tree.graph(), new PlaceNode2());
+            jf.getContentPane().add(tv, BorderLayout.CENTER);
+            jf.setVisible(true);
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
     }
 }