kenlm/lm/interpolate/merge_vocab.hh

#ifndef LM_INTERPOLATE_MERGE_VOCAB_H
#define LM_INTERPOLATE_MERGE_VOCAB_H

#include "../word_index.hh"
#include "../../util/file.hh"
#include "../../util/fixed_array.hh"

namespace lm {

class EnumerateVocab;

namespace interpolate {

class UniversalVocab;

// The combined vocabulary is enumerated with enumerate.
// Returns the size of the combined vocabulary.
// Does not take ownership of vocab_files.
WordIndex MergeVocab(util::FixedArray<int> &vocab_files, UniversalVocab &vocab, EnumerateVocab &enumerate);

}} // namespaces

#endif // LM_INTERPOLATE_MERGE_VOCAB_H