12#ifndef MLPACK_CORE_TREE_COSINE_TREE_COSINE_TREE_HPP
13#define MLPACK_CORE_TREE_COSINE_TREE_COSINE_TREE_HPP
16#include <boost/heap/priority_queue.hpp>
22class CompareCosineNode;
26typedef boost::heap::priority_queue<CosineTree*,
116 arma::vec& newBasisVector,
117 arma::vec* addBasisVector = NULL);
133 arma::vec* addBasisVector1 = NULL,
134 arma::vec* addBasisVector2 = NULL);
157 arma::vec& probabilities,
size_t numSamples);
179 size_t BinarySearch(arma::vec& cDistribution,
double value,
size_t start,
207 void L2Error(
const double error) { this->l2Error = error; }
215 void BasisVector(arma::vec& bVector) { this->basisVector = bVector; }
246 const arma::mat* dataset;
258 std::vector<size_t> indices;
260 arma::vec l2NormsSquared;
264 arma::vec basisVector;
266 size_t splitPointIndex;
272 double frobNormSquared;
bool operator()(const CosineTree *a, const CosineTree *b) const
void ConstructBasis(CosineNodeQueue &treeQueue)
Constructs the final basis matrix, after the cosine tree construction.
size_t ColumnSampleLS()
Sample a point from the Length-Squared distribution of the cosine node.
void ColumnSamplesLS(std::vector< size_t > &sampledIndices, arma::vec &probabilities, size_t numSamples)
Sample 'numSamples' points from the Length-Squared distribution of the cosine node.
CosineTree *& Parent()
Modify the pointer to the parent node.
CosineTree(const arma::mat &dataset)
CosineTree constructor for the root node of the tree.
CosineTree & operator=(CosineTree &&other)
Take ownership of the given Cosine Tree.
size_t NumColumns() const
Get number of columns of input matrix in the node.
CosineTree * Right() const
Get pointer to the right child of the node.
size_t SplitPointIndex() const
Get the column index of split point of the node.
CosineTree(CosineTree &&other)
Move the given tree.
void ModifiedGramSchmidt(CosineNodeQueue &treeQueue, arma::vec ¢roid, arma::vec &newBasisVector, arma::vec *addBasisVector=NULL)
Calculates the orthonormalization of the passed centroid, with respect to the current vector subspace...
void BasisVector(arma::vec &bVector)
Set the basis vector of the node.
arma::vec & Centroid()
Get pointer to the centroid vector.
std::vector< size_t > & VectorIndices()
Get the indices of columns in the node.
void L2Error(const double error)
Set the Monte Carlo error.
double L2Error() const
Get the Monte Carlo error.
CosineTree(const CosineTree &other)
Copy the given tree.
CosineTree(const arma::mat &dataset, const double epsilon, const double delta)
Construct the CosineTree and the basis for the given matrix, and passed 'epsilon' and 'delta' paramet...
void CalculateCosines(arma::vec &cosines)
Calculate cosines of the columns present in the node, with respect to the sampled splitting point.
CosineTree * Left() const
Get pointer to the left child of the node.
const arma::mat & GetDataset() const
Get pointer to the dataset matrix.
double FrobNormSquared() const
Get the Frobenius norm squared of columns in the node.
void GetFinalBasis(arma::mat &finalBasis)
Returns the basis of the constructed subspace.
CosineTree *& Right()
Modify the pointer to the left child of the node.
size_t BinarySearch(arma::vec &cDistribution, double value, size_t start, size_t end)
Sample a column based on the cumulative Length-Squared distribution of the cosine node,...
~CosineTree()
Clean up the CosineTree: release allocated memory (including children).
CosineTree * Parent() const
Get pointer to the parent node.
double MonteCarloError(CosineTree *node, CosineNodeQueue &treeQueue, arma::vec *addBasisVector1=NULL, arma::vec *addBasisVector2=NULL)
Estimates the squared error of the projection of the input node's matrix onto the current vector subs...
CosineTree *& Left()
Modify the pointer to the left child of the node.
void CalculateCentroid()
Calculate centroid of the columns present in the node.
CosineTree(CosineTree &parentNode, const std::vector< size_t > &subIndices)
CosineTree constructor for nodes other than the root node of the tree.
CosineTree & operator=(const CosineTree &other)
Copy the given Cosine Tree.
arma::vec & BasisVector()
Get the basis vector of the node.
void CosineNodeSplit()
This function splits the cosine node into two children based on the cosines of the columns contained ...
boost::heap::priority_queue< CosineTree *, boost::heap::compare< CompareCosineNode > > CosineNodeQueue
Linear algebra utility functions, generally performed on matrices or vectors.
The core includes that mlpack expects; standard C++ includes and Armadillo.