diff options
| -rw-r--r-- | decoder/ff_klm.cc | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/decoder/ff_klm.cc b/decoder/ff_klm.cc index a4b26f7c..7a84add7 100644 --- a/decoder/ff_klm.cc +++ b/decoder/ff_klm.cc @@ -373,15 +373,17 @@ boost::shared_ptr<FeatureFunction> KLanguageModelFactory::Create(std::string par if (!RecognizeBinary(filename.c_str(), m)) m = HASH_PROBING; switch (m) { - case HASH_PROBING: + case PROBING: return CreateModel<ProbingModel>(param); - case TRIE_SORTED: + case REST_PROBING: + return CreateModel<RestProbingModel>(param); + case TRIE: return CreateModel<TrieModel>(param); - case ARRAY_TRIE_SORTED: + case ARRAY_TRIE: return CreateModel<ArrayTrieModel>(param); - case QUANT_TRIE_SORTED: + case QUANT_TRIE: return CreateModel<QuantTrieModel>(param); - case QUANT_ARRAY_TRIE_SORTED: + case QUANT_ARRAY_TRIE: return CreateModel<QuantArrayTrieModel>(param); default: UTIL_THROW(util::Exception, "Unrecognized kenlm binary file type " << (unsigned)m); |