[mlpack-git] master: Add unfinished serialization test. (df2dac2)

gitdub at big.cc.gt.atl.ga.us gitdub at big.cc.gt.atl.ga.us
Tue Sep 29 09:33:29 EDT 2015


Repository : https://github.com/mlpack/mlpack

On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/cbeb3ea17262b7c5115247dc217e316c529249b7...f85a9b22f3ce56143943a2488c05c2810d6b2bf3

>---------------------------------------------------------------

commit df2dac2a7ea42152181cf5e805f0b70ecd205853
Author: Ryan Curtin <ryan at ratml.org>
Date:   Fri Sep 25 17:28:04 2015 -0400

    Add unfinished serialization test.


>---------------------------------------------------------------

df2dac2a7ea42152181cf5e805f0b70ecd205853
 src/mlpack/tests/serialization_test.cpp | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/mlpack/tests/serialization_test.cpp b/src/mlpack/tests/serialization_test.cpp
index 3f2d70e..061e799 100644
--- a/src/mlpack/tests/serialization_test.cpp
+++ b/src/mlpack/tests/serialization_test.cpp
@@ -24,6 +24,7 @@
 
 #include <mlpack/methods/perceptron/perceptron.hpp>
 #include <mlpack/methods/logistic_regression/logistic_regression.hpp>
+#include <mlpack/methods/neighbor_search/neighbor_search.hpp>
 
 using namespace mlpack;
 using namespace mlpack::distribution;
@@ -745,4 +746,31 @@ BOOST_AUTO_TEST_CASE(LogisticRegressionTest)
   BOOST_REQUIRE_CLOSE(lr.Lambda(), lrBinary.Lambda(), 1e-5);
 }
 
+BOOST_AUTO_TEST_CASE(AllkNNTest)
+{
+  using neighbor::AllkNN;
+  arma::mat dataset = arma::randu<arma::mat>(5, 2000);
+/*
+  AllkNN allknn(dataset, false, false);
+
+  AllkNN knnXml, knnText, knnBinary;
+
+  SerializeObjectAll(allknn, knnXml, knnText, knnBinary);
+
+  // Now run nearest neighbor and make sure the results are the same.
+  arma::mat querySet = arma::randu<arma::mat>(5, 1000);
+
+  arma::mat distances, xmlDistances, textDistances, binaryDistances;
+  arma::Mat<size_t> neighbors, xmlNeighbors, textNeighbors, binaryNeighbors;
+
+  allknn.Search(querySet, 5, neighbors, distances);
+  knnXml.Search(querySet, 5, xmlNeighbors, xmlDistances);
+  knnText.Search(querySet, 5, textNeighbors, textDistances);
+  knnBinary.Search(querySet, 5, binaryNeighbors, binaryDistances);
+
+  CheckMatrices(distances, xmlDistances, textDistances, binaryDistances);
+  CheckMatrices(neighbors, xmlNeighbors, textNeighbors, binaryNeighbors);
+*/
+}
+
 BOOST_AUTO_TEST_SUITE_END();



More information about the mlpack-git mailing list