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

Reduce KDE tests sizes to save time

Browse Source
This commit is contained in:
Roberto Hueso Gomez
2019-08-26 19:27:22 +02:00
parent ab31584138
commit d232ff3455
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/kde_test.cpp
+6 -6
View File
@@ -135,8 +135,8 @@ BOOST_AUTO_TEST_CASE(KDETreeAsArguments)
*/
BOOST_AUTO_TEST_CASE(GaussianKDEBruteForceTest)
{
arma::mat reference = arma::randu(2, 3000);
arma::mat query = arma::randu(2, 100);
arma::mat reference = arma::randu(2, 200);
arma::mat query = arma::randu(2, 60);
arma::vec bfEstimations = arma::vec(query.n_cols, arma::fill::zeros);
arma::vec treeEstimations = arma::vec(query.n_cols, arma::fill::zeros);
const double kernelBandwidth = 0.12;
@@ -169,7 +169,7 @@ BOOST_AUTO_TEST_CASE(GaussianKDEBruteForceTest)
*/
BOOST_AUTO_TEST_CASE(GaussianSingleKDEBruteForceTest)
{
arma::mat reference = arma::randu(2, 3000);
arma::mat reference = arma::randu(2, 300);
arma::mat query = arma::randu(2, 100);
arma::vec bfEstimations = arma::vec(query.n_cols, arma::fill::zeros);
arma::vec treeEstimations = arma::vec(query.n_cols, arma::fill::zeros);
@@ -249,9 +249,9 @@ BOOST_AUTO_TEST_CASE(EpanechnikovOctreeSingleKDETest)
// Brute force KDE.
EpanechnikovKernel kernel(kernelBandwidth);
BruteForceKDE<EpanechnikovKernel>(reference,
query,
bfEstimations,
kernel);
query,
bfEstimations,
kernel);
// Optimized KDE.
metric::EuclideanDistance metric;