mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-12-26 13:23:25 +03:00
29 lines
724 B
C++
29 lines
724 B
C++
#ifndef LM_ENUMERATE_VOCAB_H
|
|
#define LM_ENUMERATE_VOCAB_H
|
|
|
|
#include "lm/word_index.hh"
|
|
#include "util/string_piece.hh"
|
|
|
|
namespace lm {
|
|
|
|
/* If you need the actual strings in the vocabulary, inherit from this class
|
|
* and implement Add. Then put a pointer in Config.enumerate_vocab; it does
|
|
* not take ownership. Add is called once per vocab word. index starts at 0
|
|
* and increases by 1 each time. This is only used by the Model constructor;
|
|
* the pointer is not retained by the class.
|
|
*/
|
|
class EnumerateVocab {
|
|
public:
|
|
virtual ~EnumerateVocab() {}
|
|
|
|
virtual void Add(WordIndex index, const StringPiece &str) = 0;
|
|
|
|
protected:
|
|
EnumerateVocab() {}
|
|
};
|
|
|
|
} // namespace lm
|
|
|
|
#endif // LM_ENUMERATE_VOCAB_H
|
|
|