@Override
public Object terminate(@SuppressWarnings("deprecation") AggregationBuffer agg)
throws HiveException {
PLSAPredictAggregationBuffer myAggr = (PLSAPredictAggregationBuffer) agg;
float[] topicDistr = myAggr.get();
SortedMap<Float, Integer> sortedDistr = new TreeMap<Float, Integer>(
Collections.reverseOrder());
for (int i = 0; i < topicDistr.length; i++) {
sortedDistr.put(topicDistr[i], i);
}
List<Object[]> result = new ArrayList<Object[]>();
for (Map.Entry<Float, Integer> e : sortedDistr.entrySet()) {
Object[] struct = new Object[2];
struct[0] = new IntWritable(e.getValue().intValue()); // label
struct[1] = new FloatWritable(e.getKey().floatValue()); // probability
result.add(struct);
}
return result;
}
PLSAPredictUDAF.java 文件源码
java
阅读 28
收藏 0
点赞 0
评论 0
项目:incubator-hivemall
作者:
评论列表
文章目录