diff --git a/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp b/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp index 5ed3348aaf..fed45340e3 100644 --- a/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp +++ b/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp @@ -76,7 +76,7 @@ class RectangleTree * @param leafSize Size of each leaf in the tree; * @param maxNumChildren The maximum number of child nodes a non-leaf node may have. */ - RectangleTree(MatType& data, const size_t leafSize = 20, const size_t maxNumChildren = 5); + RectangleTree(MatType& data, const size_t leafSize = 20, const size_t maxNumChildren = 4); //TODO implement the oldFromNew stuff if applicable. diff --git a/src/mlpack/methods/neighbor_search/neighbor_search_rules.hpp b/src/mlpack/methods/neighbor_search/neighbor_search_rules.hpp index cd37e44c38..effa4ac567 100644 --- a/src/mlpack/methods/neighbor_search/neighbor_search_rules.hpp +++ b/src/mlpack/methods/neighbor_search/neighbor_search_rules.hpp @@ -22,7 +22,14 @@ class NeighborSearchRules arma::Mat& neighbors, arma::mat& distances, MetricType& metric); - + /** + * Get the distance from the query point to the reference point. + * This will update the "neighbor" matrix with the new point if appropriate + * and will track the number of base cases (number of points evaluated). + * + * @param queryIndex Index of query point. + * @param referenceIndex Index of reference point. + */ double BaseCase(const size_t queryIndex, const size_t referenceIndex); /**