Ver Fonte

Merge branch 'master' into 'master'

Master

See merge request yocj/g4b!12
KREBS-CHEVRESSON CLEMENT há 5 anos atrás
pai
commit
4a3983e929
4 ficheiros alterados com 21 adições e 7 exclusões
  1. 1 1
      .idea/misc.xml
  2. 0 4
      src/main/View.java
  3. 19 1
      src/main/Weka.java
  4. 1 1
      src/main/main.java

+ 1 - 1
.idea/misc.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" languageLevel="JDK_11" default="true" project-jdk-name="11" project-jdk-type="JavaSDK">
+  <component name="ProjectRootManager" version="2" languageLevel="JDK_11" default="true" project-jdk-name="11.0.3" project-jdk-type="JavaSDK">
     <output url="file://$PROJECT_DIR$/out" />
   </component>
 </project>

+ 0 - 4
src/main/View.java

@@ -22,10 +22,6 @@ public class View extends JFrame {
     private JPanel panimg = new JPanel();
     private ArrayList<Jul> juls = new ArrayList<>();
 
-    public static void main(String[] args){
-        new View();
-    }
-
     public View(){
         this.setTitle("Jul Project");
         this.setSize(800, 1040);

+ 19 - 1
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 {
@@ -63,8 +70,19 @@ public class Weka {
            i.setValue((Attribute)fvJulAttributes.elementAt(2), j.getCollier());
            i.setValue((Attribute)fvJulAttributes.elementAt(3), j.getChapeau());
            i.setValue((Attribute)fvJulAttributes.elementAt(4), String.valueOf(j.isChoisi()));
-           System.out.println(String.valueOf(j.isChoisi()));
            isTrainingSet.add(i);
         }
+
+        J48 tree = new J48();
+        try {
+            tree.buildClassifier(isTrainingSet);
+            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();
+        }
     }
 }

+ 1 - 1
src/main/main.java

@@ -2,6 +2,6 @@ package main;
 
 public class main {
     public static void main(String[] a) {
-
+        new View();
     }
 }