diff options
| author | Chris Dyer <cdyer@cs.cmu.edu> | 2010-12-23 14:50:41 -0600 |
|---|---|---|
| committer | Chris Dyer <cdyer@cs.cmu.edu> | 2010-12-23 14:50:41 -0600 |
| commit | 1ffcac39647bdc13e6f6ef73ade6b88d59a08101 (patch) | |
| tree | 0454153c2b5cd92d28318f834760283b0d03c237 /decoder/cdec_ff.cc | |
| parent | 68012b3700117bd21e8d62b6e5006ac25ac772c8 (diff) | |
support different types in kenlm
Diffstat (limited to 'decoder/cdec_ff.cc')
| -rw-r--r-- | decoder/cdec_ff.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/decoder/cdec_ff.cc b/decoder/cdec_ff.cc index 75591af8..686905ad 100644 --- a/decoder/cdec_ff.cc +++ b/decoder/cdec_ff.cc @@ -35,7 +35,6 @@ void register_feature_functions() { RegisterFsaDynToFF<SameFirstLetter>(); RegisterFF<LanguageModel>(); - RegisterFF<KLanguageModel>(); RegisterFF<WordPenalty>(); RegisterFF<SourceWordPenalty>(); @@ -48,6 +47,10 @@ void register_feature_functions() { #ifdef HAVE_RANDLM ff_registry.Register("RandLM", new FFFactory<LanguageModelRandLM>); #endif + ff_registry.Register("KLanguageModel", new FFFactory<KLanguageModel<lm::ngram::ProbingModel> >()); + ff_registry.Register("KLanguageModel_Sorted", new FFFactory<KLanguageModel<lm::ngram::SortedModel> >()); + ff_registry.Register("KLanguageModel_Trie", new FFFactory<KLanguageModel<lm::ngram::TrieModel> >()); + ff_registry.Register("KLanguageModel_Probing", new FFFactory<KLanguageModel<lm::ngram::ProbingModel> >()); ff_registry.Register("RuleShape", new FFFactory<RuleShapeFeatures>); ff_registry.Register("RelativeSentencePosition", new FFFactory<RelativeSentencePosition>); ff_registry.Register("LexNullJump", new FFFactory<LexNullJump>); |