diff options
author | Kenneth Heafield <github@kheafield.com> | 2012-05-16 13:27:02 -0700 |
---|---|---|
committer | Chris Dyer <cdyer@cab.ark.cs.cmu.edu> | 2012-05-26 23:04:31 -0400 |
commit | 8b123173191853466b671bbaf6f06e1c92f26095 (patch) | |
tree | b0893749d5c882d31703ebf09f7443d6ccb45e58 /decoder | |
parent | 149232c38eec558ddb1097698d1570aacb67b59f (diff) |
And actually allow ff_klm.cc to load RestProbingModel
Diffstat (limited to 'decoder')
-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); |