|
@@ -1,14 +1,12 @@
|
|
|
package main;
|
|
package main;
|
|
|
|
|
|
|
|
-import weka.core.Attribute;
|
|
|
|
|
-import weka.core.FastVector;
|
|
|
|
|
-import weka.core.Instances;
|
|
|
|
|
|
|
+import weka.core.*;
|
|
|
|
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.ArrayList;
|
|
|
|
|
|
|
|
public class Weka {
|
|
public class Weka {
|
|
|
|
|
|
|
|
- public static void traitement() {
|
|
|
|
|
|
|
+ public static void traitement(ArrayList<Jul> juls) {
|
|
|
// attributs nominaux avec leurs valeurs
|
|
// attributs nominaux avec leurs valeurs
|
|
|
FastVector fvLunettes = new FastVector(3);
|
|
FastVector fvLunettes = new FastVector(3);
|
|
|
fvLunettes.addElement("vue");
|
|
fvLunettes.addElement("vue");
|
|
@@ -38,8 +36,8 @@ public class Weka {
|
|
|
// attribut cible
|
|
// attribut cible
|
|
|
|
|
|
|
|
FastVector fvCible = new FastVector(2);
|
|
FastVector fvCible = new FastVector(2);
|
|
|
- fvCible.addElement("oui");
|
|
|
|
|
- fvCible.addElement("non");
|
|
|
|
|
|
|
+ fvCible.addElement("true");
|
|
|
|
|
+ fvCible.addElement("false");
|
|
|
|
|
|
|
|
Attribute cible = new Attribute("cible", fvCible);
|
|
Attribute cible = new Attribute("cible", fvCible);
|
|
|
|
|
|
|
@@ -54,9 +52,19 @@ public class Weka {
|
|
|
|
|
|
|
|
|
|
|
|
|
// création entrainement
|
|
// création entrainement
|
|
|
- int nElem = 0;
|
|
|
|
|
- Instances isTrainingSet = new Instances("Jul", fvJulAttributes, nElem);
|
|
|
|
|
-
|
|
|
|
|
- isTrainingSet.setClassIndex(4);
|
|
|
|
|
|
|
+ Instances isTrainingSet = new Instances("Jul", fvJulAttributes, juls.size());
|
|
|
|
|
+
|
|
|
|
|
+ isTrainingSet.setClassIndex(isTrainingSet.numAttributes() - 1);
|
|
|
|
|
+
|
|
|
|
|
+ for (Jul j : juls) {
|
|
|
|
|
+ Instance i = new DenseInstance(isTrainingSet.numAttributes());
|
|
|
|
|
+ i.setValue((Attribute)fvJulAttributes.elementAt(0), j.getLunettes());
|
|
|
|
|
+ i.setValue((Attribute)fvJulAttributes.elementAt(1), j.getMasque());
|
|
|
|
|
+ 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);
|
|
|
|
|
+ }
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|