0

我想用 Weka 来解决我的分类问题。我有一组训练数据实例。假设数据如下所示：

@relation Relation1

@attribute att1 {val11, val12}
@attribute att2 {val21, val22}
@attribute class {class1, class2, class3}

@data
val11, val21, class1
val11, val22, class2
val12, val21, class3

在我的代码中，我从文件中读取了训练集。我训练 J48 树并尝试对实例进行分类。但是，我不知道如何解释分类结果。

我的代码如下：

try {
    DataSource source = new DataSource("trainingset.arff");
    Instances data = source.getDataSet();
    if (data.classIndex() == -1) {
        data.setClassIndex(data.numAttributes() - 1);
    }

    Instance xyz = new Instance(data.numAttributes());
    xyz.setDataset(data);
    xyz.setValue(data.attribute(0), "val11");
    xyz.setValue(data.attribute(1), "val21");

    String[] options = new String[1];
    options[0] = "-U"; // unpruned tree
    J48 tree = new J48(); // new instance of tree
    tree.setOptions(options); // set the options
    tree.buildClassifier(data); // build classifier

    double[] distributionForInstance = tree.distributionForInstance(xyz);
    System.out.println(distributionForInstance[0]);
    System.out.println(distributionForInstance[1]);
    System.out.println(distributionForInstance[2]);

} catch (Exception e) {
    // TODO Auto-generated catch block
    e.printStackTrace();
}

作为输出，我得到：

0.3333333333333333
0.3333333333333333
0.3333333333333333

我还尝试了其他对实例进行分类的方法：

double classifyInstance = tree.classifyInstance(xyz);
            System.out.println(classifyInstance);

在这种情况下，输出为：

0.0

您能解释一下我应该如何解释 distributionForInstance 和分类实例方法的输出吗？我的目标是能够创建分类器，它会告诉我给定实例属于哪个类。

4

2 回答 2

0

看看javadoc。该distributionForInstance方法返回一个具有类成员概率的数组（实例在第一类中的第一个元素概率等）并classifyInstance返回类（作为 ID - 考虑类标签数组的索引）。

于 2012-12-28T18:19:46.033 回答
0

使用value方法Attribute获取类标签：

double classifyInstance = tree.classifyInstance(xyz);
String classStr = data.classAttribute().value(classifyInstance);
于 2013-01-02T07:47:08.867 回答