KmTree Class Reference

Detailed Description

Definition at line 35 of file KmTree.h.

#include <math/kmeans++/KmTree.h>

Classes
struct	Node

Public Member Functions
	KmTree (int n, int d, Scalar *points)
	~KmTree ()
Scalar	DoKMeansStep (int k, Scalar *centers, int *assignment) const
Scalar	SeedKMeansPlusPlus (int k, Scalar *centers) const

Private Member Functions
Node *	BuildNodes (Scalar *points, int first_index, int last_index, char **next_node_data)
Scalar	GetNodeCost (const Node *node, Scalar *center) const
Scalar	DoKMeansStepAtNode (const Node *node, int k, int *candidates, Scalar *centers, Scalar *sums, int *counts, int *assignment) const
bool	ShouldBePruned (Scalar *box_median, Scalar *box_radius, Scalar *centers, int best_index, int test_index) const
void	SeedKmppSetClusterIndex (const Node *node, int index) const
Scalar	SeedKmppUpdateAssignment (const Node *node, int new_cluster, Scalar *centers, Scalar *dist_sq) const

Private Attributes
int	n_
int	d_
Scalar *	points_
Node *	top_node_
char *	node_data_
int *	point_indices_

Constructor & Destructor Documentation

KmTree()

KmTree::KmTree	(	int	n,
		int	d,
		Scalar *	points
	)

Definition at line 19 of file KmTree.cpp.

References BuildNodes(), d_, KM_ASSERT, node_data_, point_indices_, PointAllocate(), PointCopy(), PointFree(), and top_node_.

~KmTree()

KmTree::~KmTree

(

)

Definition at line 49 of file KmTree.cpp.

References node_data_, and point_indices_.

Member Function Documentation

BuildNodes()

KmTree::Node * KmTree::BuildNodes ( Scalar *  points, int  first_index, int  last_index, char **  next_node_data  )

private

Definition at line 96 of file KmTree.cpp.

References d_, KmTree::Node::first_point_index, GetNodeCost(), KM_ASSERT, KmTree::Node::lower_node, KmTree::Node::median, KmTree::Node::num_points, KmTree::Node::opt_cost, point_indices_, PointAdd(), PointAllocate(), PointCopy(), PointFree(), PointScale(), KmTree::Node::radius, KmTree::Node::sum, and KmTree::Node::upper_node.

Referenced by KmTree().

DoKMeansStep()

Scalar KmTree::DoKMeansStep	(	int	k,
		Scalar *	centers,
		int *	assignment
	)		const

Definition at line 54 of file KmTree.cpp.

References d_, DoKMeansStepAtNode(), KM_ASSERT, mrpt::system::os::memcpy(), PointAllocate(), PointCopy(), PointFree(), PointScale(), and top_node_.

Referenced by RunKMeansOnce().

DoKMeansStepAtNode()

Scalar KmTree::DoKMeansStepAtNode ( const Node *  node, int  k, int *  candidates, Scalar *  centers, Scalar *  sums, int *  counts, int *  assignment  ) const

private

Definition at line 218 of file KmTree.cpp.

References d_, KmTree::Node::first_point_index, GetNodeCost(), KM_ASSERT, KmTree::Node::lower_node, KmTree::Node::median, KmTree::Node::num_points, point_indices_, PointAdd(), PointDistSq(), KmTree::Node::radius, ShouldBePruned(), KmTree::Node::sum, and KmTree::Node::upper_node.

Referenced by DoKMeansStep().

GetNodeCost()

Scalar KmTree::GetNodeCost ( const Node *  node, Scalar *  center  ) const

private

Definition at line 202 of file KmTree.cpp.

References d_, KmTree::Node::num_points, KmTree::Node::opt_cost, and KmTree::Node::sum.

Referenced by BuildNodes(), DoKMeansStepAtNode(), and SeedKmppUpdateAssignment().

SeedKMeansPlusPlus()

Scalar KmTree::SeedKMeansPlusPlus	(	int	k,
		Scalar *	centers
	)		const

Definition at line 292 of file KmTree.cpp.

References d_, GetRandom(), KM_ASSERT, mrpt::system::os::memcpy(), n_, point_indices_, PointDistSq(), points_, SeedKmppSetClusterIndex(), SeedKmppUpdateAssignment(), and top_node_.

Referenced by RunKMeansPlusPlus().

SeedKmppSetClusterIndex()

void KmTree::SeedKmppSetClusterIndex ( const Node *  node, int  index  ) const

private

Definition at line 332 of file KmTree.cpp.

References KmTree::Node::kmpp_cluster_index, KmTree::Node::lower_node, and KmTree::Node::upper_node.

Referenced by SeedKMeansPlusPlus(), and SeedKmppUpdateAssignment().

SeedKmppUpdateAssignment()

Scalar KmTree::SeedKmppUpdateAssignment ( const Node *  node, int  new_cluster, Scalar *  centers, Scalar *  dist_sq  ) const

private

Definition at line 340 of file KmTree.cpp.

References d_, KmTree::Node::first_point_index, GetNodeCost(), KmTree::Node::kmpp_cluster_index, KmTree::Node::lower_node, KmTree::Node::median, KmTree::Node::num_points, point_indices_, PointDistSq(), points_, KmTree::Node::radius, SeedKmppSetClusterIndex(), ShouldBePruned(), and KmTree::Node::upper_node.

Referenced by SeedKMeansPlusPlus().

ShouldBePruned()

bool KmTree::ShouldBePruned ( Scalar *  box_median, Scalar *  box_radius, Scalar *  centers, int  best_index, int  test_index  ) const

private

Definition at line 273 of file KmTree.cpp.

References d_.

Referenced by DoKMeansStepAtNode(), and SeedKmppUpdateAssignment().

Member Data Documentation

d_

int KmTree::d_

private

Definition at line 78 of file KmTree.h.

Referenced by BuildNodes(), DoKMeansStep(), DoKMeansStepAtNode(), GetNodeCost(), KmTree(), SeedKMeansPlusPlus(), SeedKmppUpdateAssignment(), and ShouldBePruned().

n_

int KmTree::n_

private

Definition at line 78 of file KmTree.h.

Referenced by SeedKMeansPlusPlus().

node_data_

char* KmTree::node_data_

private

Definition at line 81 of file KmTree.h.

Referenced by KmTree(), and ~KmTree().

point_indices_

int* KmTree::point_indices_

private

Definition at line 82 of file KmTree.h.

Referenced by BuildNodes(), DoKMeansStepAtNode(), KmTree(), SeedKMeansPlusPlus(), SeedKmppUpdateAssignment(), and ~KmTree().

points_

Scalar* KmTree::points_

private

Definition at line 79 of file KmTree.h.

Referenced by SeedKMeansPlusPlus(), and SeedKmppUpdateAssignment().

top_node_

Node* KmTree::top_node_

private

Definition at line 80 of file KmTree.h.

Referenced by DoKMeansStep(), KmTree(), and SeedKMeansPlusPlus().