diff options
author | Kenneth Heafield <github@kheafield.com> | 2013-11-05 04:36:24 +0400 |
---|---|---|
committer | Kenneth Heafield <github@kheafield.com> | 2013-11-05 04:36:24 +0400 |
commit | 25128843bcc123e18d0298649fa88c2e6d1a6ec7 (patch) | |
tree | 6ec6f9abbd5d65ec7f01376fa49cfa1f66b4ace4 | |
parent | 67560d5d3b7efa7d5f77a4b774e6a39ee5a963a6 (diff) |
Pull out my own method call for queries
-rw-r--r-- | src/neuralLM.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/neuralLM.h b/src/neuralLM.h index a345ea9..104b2cc 100644 --- a/src/neuralLM.h +++ b/src/neuralLM.h @@ -131,6 +131,11 @@ public: return shared->output_vocab.lookup_word(word); } + Eigen::Matrix<int,Eigen::Dynamic,1> &staging_ngram() { return ngram; } + double lookup_from_staging() { + return lookup_ngram(ngram); + } + template <typename Derived> double lookup_ngram(const Eigen::MatrixBase<Derived> &ngram) { |