1

我正在使用 weka 构建随机森林模型。我的数据存储在 MySQL 数据库中。我找不到直接从数据库创建 weka 数据集(“实例”对象)的方法(至少不是一个简单的对象),所以我查询数据库并使用此代码将数据操作到 weka 数据集(实例) :

    List<MetadataRecord> metadata = acquireMetadata(); // Loading from DB

    int datasetSize = metadata.size();
    int numFeatures = MetadataRecord.FEATURE_NUM;  // Currently set to 14

    ArrayList<Attribute> atts = new ArrayList<Attribute>();
    List<Instance> instances = new ArrayList<Instance>();
    for (int feature = 0; feature < numFeatures; feature++) {
        Attribute current = new Attribute("Attribute" + feature, feature);
        if (feature == 0) {
            for (int obj = 0; obj < datasetSize; obj++) {
                instances.add(new SparseInstance(numFeatures));
            }
        }

        for (int obj = 0; obj < datasetSize; obj++) {
            MetadataRecord record = metadata.get(obj);
            Instance inst = instances.get(obj);
            switch (feature) {
            case 0:
                inst.setValue(current, record.labelId);
                break;
            case 1:
                inst.setValue(current, record.isSecured ? 2 : 1);
                break;
            case 2:
                inst.setValue(current, record.pageCount);
                break;
                // Spared cases 3-13...
            }
        }
        atts.add(current);
    }

    Instances newDataset = new Instances("Metadata", atts, instances.size());

    for (Instance inst : instances) {
        newDataset.add(inst);
    }
    newDataset.setClassIndex(0);

大多数数据以“数字”形式输入,而我需要一些特征(第一和第二)是分类的(或“名义”,根据 weka 术语)。我尝试使用过滤器将它们转换为标称:

    NumericToNominal nomFilter = new NumericToNominal();
    nomFilter.setAttributeIndicesArray(new int[] { 0, 1 });
    nomFilter.setInputFormat(newDataset);
    newDataset = Filter.useFilter(newDataset, nomFilter);

这很好用,但令人惊讶的是,在调试数据集时,一些数据丢失了!

应用过滤器之前:

@attribute Attribute0 numeric
@attribute Attribute1 numeric
@attribute Attribute2 numeric
// Spared the other 10 Attributes
@data
{0 1005,1 1,2 19,3 1123,4 7,5 25,6 0.66,7 49,8 2892.21,9 5.32,10 22.63,11 0.4,12 48.95,13 5.29}

应用过滤器后:

@attribute Attribute0 {0,2,3,4,5,6,7,9,11,12,18,22,23,24,25,35,36,39,40,45,51,56,60,67,68,69,78,79,83,84,85,88,94,98,126,127,128,1001,1003,1004,1005,1007,1008,1009,1012,1013,1017,1018,1019,1022}
@attribute Attribute1 {1,2}
@attribute Attribute2 numeric
// Spared the other 10 Attributes
@data
{0 1005,2 19,3 1123,4 7,5 25,6 0.66,7 49,8 2892.21,9 5.32,10 22.63,11 0.4,12 48.95,13 5.29}

为什么我失去了第二个属性的值?

4

1 回答 1

1

该功能不会丢失,只是没有明确包含在您的输出中,因为它是sparse format。看看ARFF

稀疏 ARFF 文件与 ARFF 文件非常相似,但没有明确表示值为 0 的数据

稀疏 ARFF 文件具有相同的标题(即@relation 和@attribute 标记),但数据部分不同。而不是按顺序表示每个值,如下所示:

@data
0, X, 0, Y, "class A"
0, 0, W, 0, "class B"

非零属性由属性编号明确标识,并说明它们的值,如下所示:

@data
{1 X, 3 Y, 4 "class A"}
{2 W, 4 "class B"}

每个实例都用大括号括起来,每个条目的格式为: 其中 index 是属性索引(从 0 开始)。

请注意,稀疏实例中的省略值是 0,它们不是“缺失”值!如果某个值未知,则必须用问号 (?) 明确表示它。

尤其是最后一句话很重要。您Attribute1有两个可能的值,1 和 2。由于现在是标称值,因此值 1 的索引为 0。索引为 0 的值被省略。

再说一遍:这只是内存中的表示,当您将其打印到文件或屏幕时。您的数据集的实际内容没有改变。

于 2015-08-04T15:37:53.660 回答