mahout之MinK

版本依赖:

<dependency>
	<groupId>org.apache.mahout</groupId>
	<artifactId>mahout-core</artifactId>
	<version>0.6</version>
</dependency>

MinK用于获取有序的Top K,其内部实现采用Queue,使用详情如下:

MinK<JsonVo> minK = new MinK<JsonVo>(20, new Comparator<JsonVo>() {

	@Override
	public int compare(JsonVo o1, JsonVo o2) {
		return 0;
	}

});
for (int i = 0; i < 10; i++) {
	JsonVo tmp = new JsonVo(RandomStringUtils.random(2,"abcd"), RandomStringUtils.random(3,"abcd"));
	minK.offer(tmp);
}
System.out.println(minK.retrieve().size());

其中构造函数有大小(K)和Comparator接口

其offer执行如下:

if (queue.size() < k) {
	queue.add(item);
} else if (queueingComparator.compare(item, queue.peek()) > 0) {
	queue.add(item);
	queue.poll();
}

猜你喜欢

转载自snv.iteye.com/blog/1858901