Update document frequency from hashmap to treemap

2 files changed, 8 insertions, 4 deletions

diff --git a/linguistics/src/main/java/com/yahoo/language/significance/impl/DefaultSignificanceModel.java b/linguistics/src/main/java/com/yahoo/language/significance/impl/DefaultSignificanceModel.java
index 3244b8373ad..6e024c3025e 100644
--- a/linguistics/src/main/java/com/yahoo/language/significance/impl/DefaultSignificanceModel.java
+++ b/linguistics/src/main/java/com/yahoo/language/significance/impl/DefaultSignificanceModel.java
@@ -8,6 +8,8 @@ import com.yahoo.language.significance.SignificanceModel;
 import java.io.IOException;
 import java.nio.file.Path;
 import java.util.HashMap;
+import java.util.Map;
+import java.util.TreeMap;
 
 /**
  *
@@ -15,7 +17,7 @@ import java.util.HashMap;
  */
 public class DefaultSignificanceModel implements SignificanceModel {
     private final long corpusSize;
-    private final HashMap<String, Long> frequencies;
+    private final Map<String, Long> frequencies;
 
     private String id;
 
diff --git a/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java b/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java
index 9b7cbae834a..34e73e1b547 100644
--- a/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java
+++ b/linguistics/src/main/java/com/yahoo/language/significance/impl/DocumentFrequencyFile.java
@@ -7,6 +7,8 @@ import com.fasterxml.jackson.annotation.JsonInclude;
 import com.fasterxml.jackson.annotation.JsonProperty;
 
 import java.util.HashMap;
+import java.util.Map;
+import java.util.TreeMap;
 
 /**
  *
@@ -19,13 +21,13 @@ public class DocumentFrequencyFile {
 
     private final long documentCount;
 
-    private final HashMap<String, Long> frequencies;
+    private final Map<String, Long> frequencies;
 
     @JsonCreator
     public DocumentFrequencyFile(
             @JsonProperty("description") String description,
             @JsonProperty("document-count") long documentCount,
-            @JsonProperty("document-frequencies") HashMap<String, Long> frequencies) {
+            @JsonProperty("document-frequencies") Map<String, Long> frequencies) {
         this.description = description;
         this.documentCount = documentCount;
         this.frequencies = frequencies;
@@ -38,5 +40,5 @@ public class DocumentFrequencyFile {
     public long documentCount() { return documentCount; }
 
     @JsonProperty("document-frequencies")
-    public HashMap<String, Long> frequencies() { return frequencies; }
+    public Map<String, Long> frequencies() { return frequencies; }
 }