diff options
Diffstat (limited to 'moses/src')
-rw-r--r-- | moses/src/LM/Ken.cpp | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/moses/src/LM/Ken.cpp b/moses/src/LM/Ken.cpp index af2c11774..28d0534cd 100644 --- a/moses/src/LM/Ken.cpp +++ b/moses/src/LM/Ken.cpp @@ -328,15 +328,17 @@ LanguageModel *ConstructKenLM(const std::string &file, ScoreIndexManager &manage lm::ngram::ModelType model_type; if (lm::ngram::RecognizeBinary(file.c_str(), model_type)) { switch(model_type) { - case lm::ngram::HASH_PROBING: + case lm::ngram::PROBING: return new LanguageModelKen<lm::ngram::ProbingModel>(file, manager, factorType, lazy); - case lm::ngram::TRIE_SORTED: + case lm::ngram::REST_PROBING: + return new LanguageModelKen<lm::ngram::RestProbingModel>(file, manager, factorType, lazy); + case lm::ngram::TRIE: return new LanguageModelKen<lm::ngram::TrieModel>(file, manager, factorType, lazy); - case lm::ngram::QUANT_TRIE_SORTED: + case lm::ngram::QUANT_TRIE: return new LanguageModelKen<lm::ngram::QuantTrieModel>(file, manager, factorType, lazy); - case lm::ngram::ARRAY_TRIE_SORTED: + case lm::ngram::ARRAY_TRIE: return new LanguageModelKen<lm::ngram::ArrayTrieModel>(file, manager, factorType, lazy); - case lm::ngram::QUANT_ARRAY_TRIE_SORTED: + case lm::ngram::QUANT_ARRAY_TRIE: return new LanguageModelKen<lm::ngram::QuantArrayTrieModel>(file, manager, factorType, lazy); default: std::cerr << "Unrecognized kenlm model type " << model_type << std::endl; |