diff options
author | Paul Baltescu <pauldb89@gmail.com> | 2013-11-26 16:08:41 +0000 |
---|---|---|
committer | Paul Baltescu <pauldb89@gmail.com> | 2013-11-26 16:17:45 +0000 |
commit | 7b7167248cd605c9f5bd91b3c87e2826d2336a9f (patch) | |
tree | 608c4cf0435225b44a9d34d8f6a7d75cd841ac5a /extractor/data_array.cc | |
parent | 3ece9d5cf8c58edd55da35ac96f1a5869d7d6608 (diff) | |
parent | 0d1f0af81ad2a368bc3216451a87111be8b9f6f5 (diff) |
Merge branch 'master' of github.com:pauldb89/cdec
Diffstat (limited to 'extractor/data_array.cc')
-rw-r--r-- | extractor/data_array.cc | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/extractor/data_array.cc b/extractor/data_array.cc index ac0493fd..9612aa8a 100644 --- a/extractor/data_array.cc +++ b/extractor/data_array.cc @@ -90,6 +90,18 @@ string DataArray::GetWordAtIndex(int index) const { return id2word[data[index]]; } +vector<int> DataArray::GetWordIds(int index, int size) const { + return vector<int>(data.begin() + index, data.begin() + index + size); +} + +vector<string> DataArray::GetWords(int start_index, int size) const { + vector<string> words; + for (int word_id: GetWordIds(start_index, size)) { + words.push_back(id2word[word_id]); + } + return words; +} + int DataArray::GetSize() const { return data.size(); } |