diff options
Diffstat (limited to 'src/neuralLM.h')
-rw-r--r-- | src/neuralLM.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/neuralLM.h b/src/neuralLM.h index f0eebd8..a9a0dfe 100644 --- a/src/neuralLM.h +++ b/src/neuralLM.h @@ -80,7 +80,7 @@ class neuralLM : public neuralNetwork, graehl::replace_digits return vocab->lookup_word(slice); } - double lookup_ngram(const int *ngram_a, int n) + user_data_t lookup_ngram(const int *ngram_a, int n) { Eigen::Matrix<int,Eigen::Dynamic,1> ngram(m->ngram_size); for (int i=0; i<m->ngram_size; ++i) @@ -100,13 +100,13 @@ class neuralLM : public neuralNetwork, graehl::replace_digits return neuralNetwork::lookup_ngram(ngram); } - double lookup_ngram(const std::vector<int> &ngram_v) + user_data_t lookup_ngram(const std::vector<int> &ngram_v) { return lookup_ngram(ngram_v.data(), ngram_v.size()); } template <typename Derived> - double lookup_ngram(const Eigen::MatrixBase<Derived> &ngram) + user_data_t lookup_ngram(const Eigen::MatrixBase<Derived> &ngram) { return neuralNetwork::lookup_ngram(ngram); } |