Weka中实例的分类

时间:2012-03-08 11:22:25

标签: c# classification weka

我正在尝试在我的C#应用​​程序中使用Weka。我已经使用IKVM将Java部件引入我的.NET应用程序。这看起来效果很好。但是,对于Weka的API,我感到很茫然。 做什么如果实例在我的应用程序中以编程方式传递并且不作为ARFF文件提供,我会对它进行分类。

基本上,我正在尝试使用Weka的分类器集成一个简单的共同参考分析。我已经在Weka中直接构建了分类模型,并将其保存到磁盘上,我的.NET应用程序从中打开它并使用Weka的IKVM端口来预测类值。

这是我到目前为止所得到的:

    // This is the "entry" method for the classification method
    public IEnumerable<AttributedTokenDecorator> Execute(IEnumerable<TokenPair> items)
    {
        TokenPair[] pairs = items.ToArray();
        Classifier model = ReadModel(); // reads the Weka generated model
        FastVector fv = CreateFastVector(pairs);

        Instances instances = new Instances("licora", fv, pairs.Length);
        CreateInstances(instances, pairs);

        for(int i = 0; i < instances.numInstances(); i++)
        {
            Instance instance = instances.instance(i);
            double classification = model.classifyInstance(instance); // array index out of bounds?

            if(AsBoolean(classification))
                MakeCoreferent(pairs[i]);
        }

        throw new NotImplementedException(); // TODO
    }

    // This is a helper method to create instances from the internal model files
    private static void CreateInstances(Instances instances, IEnumerable<TokenPair> pairs)
    {
        instances.setClassIndex(instances.numAttributes() - 1);
        foreach(var pair in pairs)
        {
            var instance = new Instance(instances.numAttributes());
            instance.setDataset(instances);
            for (int i = 0; i < instances.numAttributes(); i++)
            {
                var attribute = instances.attribute(i);
                if (pair.Features.ContainsKey(attribute.name()) && pair.Features[attribute.name()] != null)
                {
                    var value = pair.Features[attribute.name()];
                    if (attribute.isNumeric()) instance.setValue(attribute, Convert.ToDouble(value));
                    else instance.setValue(attribute, value.ToString());
                }
                else
                {
                    instance.setMissing(attribute);
                }
            }
            //instance.setClassMissing();
            instances.add(instance);
        }
    }

    // This creates the data set's attributes vector
    private FastVector CreateFastVector(TokenPair[] pairs)
    {
        var fv = new FastVector();

        foreach (var attribute in _features)
        {
            Attribute att;
            if (attribute.Type.Equals(ArffType.Nominal))
            {
                var values = new FastVector();
                ExtractValues(values, pairs, attribute.FeatureName);
                att = new Attribute(attribute.FeatureName, values);
            }
            else
                att = new Attribute(attribute.FeatureName);

            fv.addElement(att);
        }

        {
            var classValues = new FastVector(2);
            classValues.addElement("0");
            classValues.addElement("1");
            var classAttribute = new Attribute("isCoref", classValues);
            fv.addElement(classAttribute);
        }

        return fv;
    }

    // This extracts observed values for nominal attributes
    private static void ExtractValues(FastVector values, IEnumerable<TokenPair> pairs, string featureName)
    {
        var strings = (from x in pairs
                       where x.Features.ContainsKey(featureName) && x.Features[featureName] != null
                       select x.Features[featureName].ToString())
            .Distinct().ToArray();

        foreach (var s in strings)
            values.addElement(s);
    }

    private Classifier ReadModel()
    {
        return (Classifier) SerializationHelper.read(_model);
    }

    private static bool AsBoolean(double classifyInstance)
    {
        return classifyInstance >= 0.5;
    }

出于某种原因,当我致电IndexOutOfRangeException时,Weka会抛出model.classifyInstance(instance)。我不知道为什么,也不能想出如何纠正这个问题。

我希望有人知道我哪里出错了。我发现Weka的唯一文档依赖于ARFF文件进行预测,我真的不想去那里。

1 个答案:

答案 0 :(得分:0)

由于一些奇怪的原因,这个例外是由DTNB分类器引发的(我在多数投票分类模型中使用了三个)。显然,没有使用DTNB“修复”这个问题。