diff options
| author | Bjørn Christian Seime <bjorn.christian@seime.no> | 2024-05-15 14:08:53 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-15 14:08:53 +0200 |
| commit | f7e2c111075dd354b9bc6dac4edc561978c68315 (patch) | |
| tree | b91ae911504b112b8d372440bc23f141fbd974e5 /linguistics | |
| parent | bd867c50b11770e953476a394f56c08780f754ab (diff) | |
| parent | b883e6279ab24d4692326196944957f82d73c523 (diff) | |
Merge pull request #31098 from vespa-engine/marius/add-significance-model-tool
Add significance model generator cli
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); + } } |