diff options
Diffstat (limited to 'gi/posterior-regularisation/prjava/src/util/Sorters.java')
-rw-r--r-- | gi/posterior-regularisation/prjava/src/util/Sorters.java | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/gi/posterior-regularisation/prjava/src/util/Sorters.java b/gi/posterior-regularisation/prjava/src/util/Sorters.java new file mode 100644 index 00000000..836444e5 --- /dev/null +++ b/gi/posterior-regularisation/prjava/src/util/Sorters.java @@ -0,0 +1,39 @@ +package util; + +import java.util.Comparator; + +public class Sorters { + public static class sortWordsCounts implements Comparator{ + + /** + * Sorter for a pair of word id, counts. Sort ascending by counts + */ + public int compare(Object arg0, Object arg1) { + Pair<Integer,Integer> p1 = (Pair<Integer,Integer>)arg0; + Pair<Integer,Integer> p2 = (Pair<Integer,Integer>)arg1; + if(p1.second() > p2.second()){ + return 1; + }else{ + return -1; + } + } + + } + +public static class sortWordsDouble implements Comparator{ + + /** + * Sorter for a pair of word id, counts. Sort by counts + */ + public int compare(Object arg0, Object arg1) { + Pair<Integer,Double> p1 = (Pair<Integer,Double>)arg0; + Pair<Integer,Double> p2 = (Pair<Integer,Double>)arg1; + if(p1.second() < p2.second()){ + return 1; + }else{ + return -1; + } + } + + } +} |