|
|
#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); |
|
|
} |
|
|
} |
|
|
|
|
|
} |
|
|
|
