Java, Weka : NaiveBayesUpdateable : Impossible de gérer la classe numérique

Question

J'essaie d'utiliser le classificateur NaiveBayesUpdateable de Weka. Mes données contiennent à la fois des attributs nominaux et numériques :

  @relation cars
  @attribute country {FR, UK, ...}
  @attribute city {London, Paris, ...}
  @attribute car_make {Toyota, BMW, ...}
  @attribute price numeric   %% car price 
  @attribute sales numeric   %% number of cars sold

Je dois prédire le nombre de ventes (numérique !) en fonction d'autres attributs. Lorsque j'exécute :

    // Train classifier
    ArffLoader loader = new ArffLoader();
    loader.setFile(new File(trainFileName));
    Instances structure = loader.getStructure();
    structure.setClassIndex(structure.numAttributes() - 1);

    // train NaiveBayes
    NaiveBayesUpdateable nb = new NaiveBayesUpdateable();
    nb.setUseKernelEstimator(true);
    nb.buildClassifier(structure);

Je reçois l'exception :

  Exception in thread "main" weka.core.UnsupportedAttributeTypeException: weka.classifiers.bayes.NaiveBayesUpdateable: Cannot handle numeric class!
      at weka.core.Capabilities.test(Capabilities.java:954)
      at weka.core.Capabilities.test(Capabilities.java:1110)
      at weka.core.Capabilities.test(Capabilities.java:1023)
      at weka.core.Capabilities.testWithFail(Capabilities.java:1302)
      at weka.classifiers.bayes.NaiveBayes.buildClassifier(NaiveBayes.java:213)
      at foo.bar.IncrementalClassifier.trainEvalPredict(IncrementalClassifier.java:65)
      at foo.bar.IncrementalClassifier.main(IncrementalClassifier.java:36)

Comment puis-je utiliser un attribut numérique pour la classification Bayes dans Weka ?

Answer 1

Vous ne pouvez pas utiliser les classificateurs Bayes dans Weka pour les prédictions numériques. Aucun d'entre eux ne supporte cela.