diff options
Diffstat (limited to 'extractor/phrase_builder.h')
-rw-r--r-- | extractor/phrase_builder.h | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/extractor/phrase_builder.h b/extractor/phrase_builder.h new file mode 100644 index 00000000..a49af457 --- /dev/null +++ b/extractor/phrase_builder.h @@ -0,0 +1,24 @@ +#ifndef _PHRASE_BUILDER_H_ +#define _PHRASE_BUILDER_H_ + +#include <memory> +#include <vector> + +using namespace std; + +class Phrase; +class Vocabulary; + +class PhraseBuilder { + public: + PhraseBuilder(shared_ptr<Vocabulary> vocabulary); + + Phrase Build(const vector<int>& symbols); + + Phrase Extend(const Phrase& phrase, bool start_x, bool end_x); + + private: + shared_ptr<Vocabulary> vocabulary; +}; + +#endif |