java - 使用 Weka 查找实例分配到的集群

Question

我使用带有 AddCluster 过滤器的 EM 聚类器，以查看训练后将哪些实例分配给不同的集群。下面是我正在使用的代码。我无法确定我正确地应用了过滤器，但是一旦我有了新的实例，我仍然不知道如何从它们那里获取集群信息。我确定它只是一个简单的 getBlah() 调用，但我只是没有找到它。提前致谢。

public Cluster()
{
    clusterer = new EM();
    filter = new AddCluster();

    try
    {
        clusterer.setMaxIterations(100);
        clusterer.setNumClusters(20);
        filter.setClusterer(clusterer);
    }
    catch (Exception e)
    {
        e.printStackTrace();
    }
}

public void buildCluster(String fileName)
{
    try
    {
        DataSource source = new DataSource(fileName);
        inst = source.getDataSet();
        filter.setInputFormat(inst);
        inst = AddCluster.useFilter(inst, filter);
    }
    catch (Exception e)
    {
        e.printStackTrace();
    }
}

Answer 1

我认为你应该使用“字典”类。这是我的示例代码：

Enumeration clusteredInst = data_to_use.enumerateInstances();<br> 
Dictionary<Integer, ArrayList<Instance>> clusteredSamples = new ashtable<>();


while (clusteredInst.hasMoreElements()) {<br> 
            Instance ins = (Instance) clusteredInst.nextElement();<br> 
            int clusterNumb = em.clusterInstance(ins);<br> 
            ArrayList<Instance> cls = null;<br> 
            cls = clusteredSamples.get(clusterNumb);<br> 
            if (cls != null) {<br> 
                cls.add(ins);<br> 
            } else {<br> 
                cls = new ArrayList<>();<br> 
                cls.add(ins);<br> 
                //you add elements to dictionary using put method<br> 
                //put(key, value)<br> 
                clusteredSamples.put(clusterNumb, cls);<br> 
            }
        }

您还可以通过调用它的键从字典中检索您的数据。