diff options
Diffstat (limited to 'linguistics')
2 files changed, 7 insertions, 4 deletions
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 b62754ac8ad..9b7cbae834a 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 @@ -17,15 +17,14 @@ import java.util.HashMap; public class DocumentFrequencyFile { private final String description; - private final int documentCount; - + private final long documentCount; private final HashMap<String, Long> frequencies; @JsonCreator public DocumentFrequencyFile( @JsonProperty("description") String description, - @JsonProperty("document-count") int documentCount, + @JsonProperty("document-count") long documentCount, @JsonProperty("document-frequencies") HashMap<String, Long> frequencies) { this.description = description; this.documentCount = documentCount; @@ -36,7 +35,7 @@ public class DocumentFrequencyFile { public String description() { return description; } @JsonProperty("document-count") - public int documentCount() { return documentCount; } + public long documentCount() { return documentCount; } @JsonProperty("document-frequencies") public HashMap<String, Long> frequencies() { return frequencies; } diff --git a/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java b/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java index 902613379f0..94030108671 100644 --- a/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java +++ b/linguistics/src/main/java/com/yahoo/language/significance/impl/SignificanceModelFile.java @@ -45,4 +45,8 @@ public class SignificanceModelFile { @JsonProperty("languages") public HashMap<String, DocumentFrequencyFile> languages() { return languages; } + + public void addLanguage(String language, DocumentFrequencyFile documentFrequencyFile) { + languages.put(language, documentFrequencyFile); + } } |