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 | 8f65daa5bdaddaac24cea4df70049757536d6080 (patch) | |
tree | 314b9fa170ef58cbff1c24136ae56ad8219f382f /extractor/data_array.cc | |
parent | 304103565d3b79cc9c98c1ee0356a8824fc982c2 (diff) | |
parent | f528ac27dab11770f01595b043675dba2947a263 (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(); } |