MLPACK
1.0.11
|
Public Types | |
typedef NeighborSearchTraversalInfo< TreeType > | TraversalInfoType |
Convenience typedef. More... | |
Public Member Functions | |
NeighborSearchRules (const typename TreeType::Mat &referenceSet, const typename TreeType::Mat &querySet, arma::Mat< size_t > &neighbors, arma::mat &distances, MetricType &metric) | |
double | BaseCase (const size_t queryIndex, const size_t referenceIndex) |
Get the distance from the query point to the reference point. More... | |
size_t | BaseCases () const |
Get the number of base cases that have been performed. More... | |
size_t & | BaseCases () |
Modify the number of base cases that have been performed. More... | |
double | Rescore (const size_t queryIndex, TreeType &referenceNode, const double oldScore) const |
Re-evaluate the score for recursion order. More... | |
double | Rescore (TreeType &queryNode, TreeType &referenceNode, const double oldScore) const |
Re-evaluate the score for recursion order. More... | |
double | Score (const size_t queryIndex, TreeType &referenceNode) |
Get the score for recursion order. More... | |
double | Score (TreeType &queryNode, TreeType &referenceNode) |
Get the score for recursion order. More... | |
size_t | Scores () const |
Get the number of scores that have been performed. More... | |
size_t & | Scores () |
Modify the number of scores that have been performed. More... | |
const TraversalInfoType & | TraversalInfo () const |
Get the traversal info. More... | |
TraversalInfoType & | TraversalInfo () |
Modify the traversal info. More... | |
Private Member Functions | |
double | CalculateBound (TreeType &queryNode) const |
Recalculate the bound for a given query node. More... | |
void | InsertNeighbor (const size_t queryIndex, const size_t pos, const size_t neighbor, const double distance) |
Insert a point into the neighbors and distances matrices; this is a helper function. More... | |
Private Attributes | |
size_t | baseCases |
The number of base cases that have been performed. More... | |
arma::mat & | distances |
The matrix the resultant neighbor distances should be stored in. More... | |
double | lastBaseCase |
The last base case result. More... | |
size_t | lastQueryIndex |
The last query point BaseCase() was called with. More... | |
size_t | lastReferenceIndex |
The last reference point BaseCase() was called with. More... | |
MetricType & | metric |
The instantiated metric. More... | |
arma::Mat< size_t > & | neighbors |
The matrix the resultant neighbor indices should be stored in. More... | |
const TreeType::Mat & | querySet |
The query set. More... | |
const TreeType::Mat & | referenceSet |
The reference set. More... | |
size_t | scores |
The number of scores that have been performed. More... | |
TraversalInfoType | traversalInfo |
Traversal info for the parent combination; this is updated by the traversal before each call to Score(). More... | |
Definition at line 32 of file neighbor_search_rules.hpp.
typedef NeighborSearchTraversalInfo<TreeType> mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::TraversalInfoType |
Convenience typedef.
Definition at line 111 of file neighbor_search_rules.hpp.
mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::NeighborSearchRules | ( | const typename TreeType::Mat & | referenceSet, |
const typename TreeType::Mat & | querySet, | ||
arma::Mat< size_t > & | neighbors, | ||
arma::mat & | distances, | ||
MetricType & | metric | ||
) |
double mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::BaseCase | ( | const size_t | queryIndex, |
const size_t | referenceIndex | ||
) |
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).
queryIndex | Index of query point. |
referenceIndex | Index of reference point. |
|
inline |
Get the number of base cases that have been performed.
Definition at line 101 of file neighbor_search_rules.hpp.
References mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::baseCases.
|
inline |
Modify the number of base cases that have been performed.
Definition at line 103 of file neighbor_search_rules.hpp.
References mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::baseCases.
|
private |
Recalculate the bound for a given query node.
|
private |
Insert a point into the neighbors and distances matrices; this is a helper function.
queryIndex | Index of point whose neighbors we are inserting into. |
pos | Position in list to insert into. |
neighbor | Index of reference point which is being inserted. |
distance | Distance from query point to reference point. |
double mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::Rescore | ( | const size_t | queryIndex, |
TreeType & | referenceNode, | ||
const double | oldScore | ||
) | const |
Re-evaluate the score for recursion order.
A low score indicates priority for recursion, while DBL_MAX indicates that the node should not be recursed into at all (it should be pruned). This is used when the score has already been calculated, but another recursion may have modified the bounds for pruning. So the old score is checked against the new pruning bound.
double mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::Rescore | ( | TreeType & | queryNode, |
TreeType & | referenceNode, | ||
const double | oldScore | ||
) | const |
Re-evaluate the score for recursion order.
A low score indicates priority for recursion, while DBL_MAX indicates that the node should not be recursed into at all (it should be pruned). This is used when the score has already been calculated, but another recursion may have modified the bounds for pruning. So the old score is checked against the new pruning bound.
queryNode | Candidate query node to recurse into. |
referenceNode | Candidate reference node to recurse into. |
oldScore | Old score produced by Socre() (or Rescore()). |
double mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::Score | ( | const size_t | queryIndex, |
TreeType & | referenceNode | ||
) |
Get the score for recursion order.
A low score indicates priority for recursion, while DBL_MAX indicates that the node should not be recursed into at all (it should be pruned).
queryIndex | Index of query point. |
referenceNode | Candidate node to be recursed into. |
double mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::Score | ( | TreeType & | queryNode, |
TreeType & | referenceNode | ||
) |
Get the score for recursion order.
A low score indicates priority for recursionm while DBL_MAX indicates that the node should not be recursed into at all (it should be pruned).
queryNode | Candidate query node to recurse into. |
referenceNode | Candidate reference node to recurse into. |
|
inline |
Get the number of scores that have been performed.
Definition at line 106 of file neighbor_search_rules.hpp.
References mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::scores.
|
inline |
Modify the number of scores that have been performed.
Definition at line 108 of file neighbor_search_rules.hpp.
References mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::scores.
|
inline |
Get the traversal info.
Definition at line 114 of file neighbor_search_rules.hpp.
References mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::traversalInfo.
|
inline |
Modify the traversal info.
Definition at line 116 of file neighbor_search_rules.hpp.
References mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::traversalInfo.
|
private |
The number of base cases that have been performed.
Definition at line 142 of file neighbor_search_rules.hpp.
Referenced by mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::BaseCases().
|
private |
The matrix the resultant neighbor distances should be stored in.
Definition at line 129 of file neighbor_search_rules.hpp.
|
private |
The last base case result.
Definition at line 139 of file neighbor_search_rules.hpp.
|
private |
The last query point BaseCase() was called with.
Definition at line 135 of file neighbor_search_rules.hpp.
|
private |
The last reference point BaseCase() was called with.
Definition at line 137 of file neighbor_search_rules.hpp.
|
private |
The instantiated metric.
Definition at line 132 of file neighbor_search_rules.hpp.
|
private |
The matrix the resultant neighbor indices should be stored in.
Definition at line 126 of file neighbor_search_rules.hpp.
|
private |
The query set.
Definition at line 123 of file neighbor_search_rules.hpp.
|
private |
The reference set.
Definition at line 120 of file neighbor_search_rules.hpp.
|
private |
The number of scores that have been performed.
Definition at line 144 of file neighbor_search_rules.hpp.
Referenced by mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::Scores().
|
private |
Traversal info for the parent combination; this is updated by the traversal before each call to Score().
Definition at line 148 of file neighbor_search_rules.hpp.
Referenced by mlpack::neighbor::NeighborSearchRules< SortPolicy, MetricType, TreeType >::TraversalInfo().