添加到StringToNominal的选项无效

时间:2015-11-18 00:42:27

标签: java weka

问题陈述

我一直在尝试使用StringToNominal将多个Weka String属性更改为Nominal属性。

使用不带选项的过滤器可以为类属性生成所需的结果。 StringToNominal默认使用last属性,类成功转换为Nominal

StringToNominal stringFilter = new StringToNominal();
stringFilter.setInputFormat(insts);
Instances filteredInsts = Filter.useFilter(insts, stringFilter); 

但是,如果我尝试通过传递一个选项来做同样的事情,那么class属性仍然是一个String

StringToNominal stringFilter = new StringToNominal();
String[] options = new String[2];
options[0] = "-R"; //Range option
options[1] = Integer.toString(insts.classIndex()); //The class attribute index
stringFilter.setOptions(options);
stringFilter.setInputFormat(insts);
Instances filteredInstsWOpts = Filter.useFilter(insts, stringFilter);

MCVE

这是错误的MCVE。它产生输出“真,假”。正确的输出是“真实,真实”

import java.util.ArrayList;

import weka.core.Attribute;
import weka.core.DenseInstance;
import weka.core.Instance;
import weka.core.Instances;
import weka.filters.Filter;
import weka.filters.unsupervised.attribute.StringToNominal;

public class TestStringToNominal {

    public static void main(String[] args) throws Exception {
        ArrayList<Attribute> attInfo = new ArrayList<Attribute>();
        attInfo.add(new Attribute("val"));
        attInfo.add(new Attribute("class", (ArrayList<String>) null));

        Instances insts = new Instances("test instances", attInfo, 1);
        insts.setClassIndex(1);

        Instance i1 = new DenseInstance(2);
        i1.setValue(attInfo.get(0), 0);
        i1.setValue(attInfo.get(1), "first");
        insts.add(i1);

        Instance i2 = new DenseInstance(2);
        i2.setValue(attInfo.get(0), 1);
        i2.setValue(attInfo.get(1), "second");
        insts.add(i2);

        StringToNominal stringFilter = new StringToNominal();
        stringFilter.setInputFormat(insts);
        Instances filteredInsts = Filter.useFilter(insts, stringFilter);
        System.out.println(filteredInsts.classAttribute().isNominal());

        StringToNominal stringFilterWOpts = new StringToNominal();
        String[] options = new String[2];
        options[0] = "-R";
        options[1] = Integer.toString(insts.classIndex());
        stringFilterWOpts.setOptions(options);
        stringFilterWOpts.setInputFormat(insts);
        Instances filteredInstsWOpts = Filter.useFilter(insts, stringFilterWOpts);
        System.out.println(filteredInstsWOpts.classAttribute().isNominal());
    }

}

1 个答案:

答案 0 :(得分:1)

我无法在任何地方找到此文档,但Instances似乎使用从零开始的索引,而StringToNominal似乎使用基于一的索引。

更改

options[1] = Integer.toString(insts.classIndex());

options[1] = Integer.toString(insts.classIndex() + 1);

产生所需的“真实,真实”输出

这个并不是很明显的原因是StringToNominal对非String类型属性没有影响,所以在这个例子中,它作用于第0个属性,一个数字类型,没有任何明显的影响。