Repository: incubator-hivemall Updated Branches: refs/heads/master 6c4ec01a4 -> fef902811
Applied refactoring Project: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/commit/fef90281 Tree: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/tree/fef90281 Diff: http://git-wip-us.apache.org/repos/asf/incubator-hivemall/diff/fef90281 Branch: refs/heads/master Commit: fef902811c58a593e14394546e63d289e34f0616 Parents: 6c4ec01 Author: Makoto Yui <[email protected]> Authored: Thu Apr 19 16:20:27 2018 +0900 Committer: Makoto Yui <[email protected]> Committed: Thu Apr 19 16:20:56 2018 +0900 ---------------------------------------------------------------------- .../hivemall/topicmodel/ProbabilisticTopicModelBaseUDTF.java | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-hivemall/blob/fef90281/core/src/main/java/hivemall/topicmodel/ProbabilisticTopicModelBaseUDTF.java ---------------------------------------------------------------------- diff --git a/core/src/main/java/hivemall/topicmodel/ProbabilisticTopicModelBaseUDTF.java b/core/src/main/java/hivemall/topicmodel/ProbabilisticTopicModelBaseUDTF.java index b740d47..c7c6b15 100644 --- a/core/src/main/java/hivemall/topicmodel/ProbabilisticTopicModelBaseUDTF.java +++ b/core/src/main/java/hivemall/topicmodel/ProbabilisticTopicModelBaseUDTF.java @@ -465,10 +465,9 @@ public abstract class ProbabilisticTopicModelBaseUDTF extends UDTFWithOptions { final SortedMap<Float, List<String>> topicWords = model.getTopicWords(k); for (Map.Entry<Float, List<String>> e : topicWords.entrySet()) { - score.set(e.getKey()); - List<String> words = e.getValue(); - for (int i = 0; i < words.size(); i++) { - word.set(words.get(i)); + score.set(e.getKey().floatValue()); + for (String v : e.getValue()) { + word.set(v); forward(forwardObjs); } }
