apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compile KNN serialization_test

Browse Source 
Signed-off-by: Omar Shrit <omar@shrit.me>
This commit is contained in:
Omar Shrit
2020-07-11 17:11:00 +02:00
parent 279cecd61e
commit 1e0640f627
2 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/neighbor_search/neighbor_search_impl.hpp
+2 -2
View File
@@ -963,7 +963,7 @@ DualTreeTraversalType, SingleTreeTraversalType>::serialize(
delete referenceSet;
}
ar(CEREAL_NVP(referenceSet));
ar(CEREAL_POINTER(referenceSet));
ar(CEREAL_NVP(metric));
// If we are loading, set the tree to NULL and clean up memory if necessary.
@@ -984,7 +984,7 @@ DualTreeTraversalType, SingleTreeTraversalType>::serialize(
delete referenceTree;
}
ar(CEREAL_NVP(referenceTree));
ar(CEREAL_POINTER(referenceTree));
ar(CEREAL_NVP(oldFromNewReferences));
// If we are loading, set the dataset accordingly and clean up memory if
src/mlpack/tests/serialization_test.cpp
+18 -18
View File
@@ -630,31 +630,31 @@ BOOST_AUTO_TEST_CASE(LogisticRegressionTest)
BOOST_REQUIRE_CLOSE(lr.Lambda(), lrBinary.Lambda(), 1e-5);
}
// BOOST_AUTO_TEST_CASE(KNNTest)
// {
// using neighbor::KNN;
// arma::mat dataset = arma::randu<arma::mat>(5, 2000);
BOOST_AUTO_TEST_CASE(KNNTest)
{
using neighbor::KNN;
arma::mat dataset = arma::randu<arma::mat>(5, 2000);
// KNN knn(dataset, DUAL_TREE_MODE);
KNN knn(dataset, DUAL_TREE_MODE);
// KNN knnXml, knnText, knnBinary;
KNN knnXml, knnText, knnBinary;
// SerializeObjectAll(knn, knnXml, knnText, knnBinary);
SerializeObjectAll(knn, knnXml, knnText, knnBinary);
// // Now run nearest neighbor and make sure the results are the same.
// arma::mat querySet = arma::randu<arma::mat>(5, 1000);
// 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;
arma::mat distances, xmlDistances, textDistances, binaryDistances;
arma::Mat<size_t> neighbors, xmlNeighbors, textNeighbors, binaryNeighbors;
// knn.Search(querySet, 5, neighbors, distances);
// knnXml.Search(querySet, 5, xmlNeighbors, xmlDistances);
// knnText.Search(querySet, 5, textNeighbors, textDistances);
// knnBinary.Search(querySet, 5, binaryNeighbors, binaryDistances);
knn.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);
// }
CheckMatrices(distances, xmlDistances, textDistances, binaryDistances);
CheckMatrices(neighbors, xmlNeighbors, textNeighbors, binaryNeighbors);
}
BOOST_AUTO_TEST_CASE(SoftmaxRegressionTest)
{