| #include "KenOSM.h" |
|
|
| namespace Moses2 |
| { |
|
|
| OSMLM* ConstructOSMLM(const char *file, util::LoadMethod load_method) |
| { |
| lm::ngram::ModelType model_type; |
| lm::ngram::Config config; |
| config.load_method = load_method; |
| if (lm::ngram::RecognizeBinary(file, model_type)) { |
| switch(model_type) { |
| case lm::ngram::PROBING: |
| return new KenOSM<lm::ngram::ProbingModel>(file, config); |
| case lm::ngram::REST_PROBING: |
| return new KenOSM<lm::ngram::RestProbingModel>(file, config); |
| case lm::ngram::TRIE: |
| return new KenOSM<lm::ngram::TrieModel>(file, config); |
| case lm::ngram::QUANT_TRIE: |
| return new KenOSM<lm::ngram::QuantTrieModel>(file, config); |
| case lm::ngram::ARRAY_TRIE: |
| return new KenOSM<lm::ngram::ArrayTrieModel>(file, config); |
| case lm::ngram::QUANT_ARRAY_TRIE: |
| return new KenOSM<lm::ngram::QuantArrayTrieModel>(file, config); |
| default: |
| UTIL_THROW2("Unrecognized kenlm model type " << model_type); |
| } |
| } else { |
| return new KenOSM<lm::ngram::ProbingModel>(file, config); |
| } |
| } |
|
|
| } |
|
|
