From 25128843bcc123e18d0298649fa88c2e6d1a6ec7 Mon Sep 17 00:00:00 2001
From: Kenneth Heafield <github@kheafield.com>
Date: Mon, 4 Nov 2013 16:36:24 -0800
Subject: Pull out my own method call for queries

---
 src/neuralLM.h | 5 +++++
 1 file changed, 5 insertions(+)

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)
     {
-- 
cgit v1.2.3