|
arma::rowvec | CalculateCentroid (arma::mat A) const |
| Calculates the centroid of the matrix. More...
|
|
void | CreateCosineSimilarityArray (std::vector< double > &c, arma::mat A, size_t pivot) |
| Creates Cosine Similarity Array. More...
|
|
double | GetMaxSimilarity (std::vector< double > c) |
| Calculates Maximum Cosine Similarity. More...
|
|
double | GetMinSimilarity (std::vector< double > c) |
| Calculates Maximum Cosine Similarity. More...
|
|
size_t | GetPivot (arma::vec prob) |
| Calculates the Pivot for splitting. More...
|
|
void | LSSampling (arma::mat A, arma::vec &prob) |
| Length Square Sampling method for sampling rows of the matrix. More...
|
|
void | SplitData (std::vector< double > c, arma::mat &ALeft, arma::mat &Aright, arma::mat A) |
| Splits the points into the root node into children nodes. More...
|
|
Definition at line 33 of file cosine_tree_builder.hpp.
mlpack::tree::CosineTreeBuilder::CosineTreeBuilder |
( |
| ) |
|
mlpack::tree::CosineTreeBuilder::~CosineTreeBuilder |
( |
| ) |
|
arma::rowvec mlpack::tree::CosineTreeBuilder::CalculateCentroid |
( |
arma::mat |
A | ) |
const |
|
private |
Calculates the centroid of the matrix.
- Parameters
-
A | Matrix for which the centroid has to be calculated |
void mlpack::tree::CosineTreeBuilder::CreateCosineSimilarityArray |
( |
std::vector< double > & |
c, |
|
|
arma::mat |
A, |
|
|
size_t |
pivot |
|
) |
| |
|
private |
Creates Cosine Similarity Array.
- Parameters
-
c | Array of Cosine Similarity |
A | All points |
pivot | pivot point |
void mlpack::tree::CosineTreeBuilder::CTNode |
( |
arma::mat |
A, |
|
|
CosineTree & |
root |
|
) |
| |
Creates a new cosine tree node.
- Parameters
-
A | Data for constructing the node |
root | Reference to the constructed node |
Splits a cosine tree node.
- Parameters
-
root | Node to be split |
right | reference to the right child |
left | reference to the left child |
double mlpack::tree::CosineTreeBuilder::GetMaxSimilarity |
( |
std::vector< double > |
c | ) |
|
|
private |
Calculates Maximum Cosine Similarity.
- Parameters
-
c | Array of Cosine Similarities |
double mlpack::tree::CosineTreeBuilder::GetMinSimilarity |
( |
std::vector< double > |
c | ) |
|
|
private |
Calculates Maximum Cosine Similarity.
- Parameters
-
c | Array of Cosine Similarities |
size_t mlpack::tree::CosineTreeBuilder::GetPivot |
( |
arma::vec |
prob | ) |
|
|
private |
Calculates the Pivot for splitting.
- Parameters
-
prob | Probability for a point to act as the pivot |
void mlpack::tree::CosineTreeBuilder::LSSampling |
( |
arma::mat |
A, |
|
|
arma::vec & |
prob |
|
) |
| |
|
private |
Length Square Sampling method for sampling rows of the matrix.
- Parameters
-
A | Matrix for which probabilities are calculated |
prob | Reference to the probability vector |
void mlpack::tree::CosineTreeBuilder::SplitData |
( |
std::vector< double > |
c, |
|
|
arma::mat & |
ALeft, |
|
|
arma::mat & |
Aright, |
|
|
arma::mat |
A |
|
) |
| |
|
private |
Splits the points into the root node into children nodes.
- Parameters
-
c | Array of Cosin Similarities |
ALeft | Matrix for storing the points in Left Node |
ARight | Matrix for storing the points in Right Node |
A | All points |
The documentation for this class was generated from the following file: