From b44979b736dbea1a1b5cc31bc935ea541855ea98 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Sun, 18 Oct 2015 07:35:19 -0400
Subject: [PATCH] Add access to the reference set.

---
 src/mlpack/methods/neighbor_search/neighbor_search.hpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/mlpack/methods/neighbor_search/neighbor_search.hpp b/src/mlpack/methods/neighbor_search/neighbor_search.hpp
index ff01c88c68..56d9841420 100644
--- a/src/mlpack/methods/neighbor_search/neighbor_search.hpp
+++ b/src/mlpack/methods/neighbor_search/neighbor_search.hpp
@@ -229,6 +229,9 @@ class NeighborSearch
   //! Modify whether or not search is done in single-tree mode.
   bool& SingleMode() { return singleMode; }
 
+  //! Access the reference dataset.
+  const MatType& ReferenceSet() const { return *referenceSet; }
+
   //! Serialize the NeighborSearch model.
   template<typename Archive>
   void Serialize(Archive& ar, const unsigned int /* version */);