22 #ifndef __MLPACK_CORE_TREE_COSINE_TREE_COSINE_TREE_BUILDER_HPP
23 #define __MLPACK_CORE_TREE_COSINE_TREE_COSINE_TREE_BUILDER_HPP
28 using namespace mlpack::tree;
43 void LSSampling(arma::mat A, arma::vec& prob);
50 arma::rowvec CalculateCentroid(arma::mat A)
const;
57 size_t GetPivot(arma::vec prob);
67 void SplitData(std::vector<double> c, arma::mat& ALeft,
68 arma::mat& Aright, arma::mat A);
77 void CreateCosineSimilarityArray(std::vector<double>& c,
78 arma::mat A,
size_t pivot);
85 double GetMaxSimilarity(std::vector<double> c);
92 double GetMinSimilarity(std::vector<double> c);
121 #include "cosine_tree_builder_impl.hpp"