apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Mark parent as pruned if its children are pruned.

Browse Source
This commit is contained in:
Ryan Curtin
2015-03-12 15:42:45 -04:00
parent cc0aec0764
commit 0968f5374f
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/kmeans/dtnn_kmeans_impl.hpp
+5 -4
View File
@@ -369,12 +369,13 @@ oldFromNewCentroids[assignments(0, node.Point(i - 1))] << ".\n";
{
// The node isn't owned by a single cluster. But if it has no points and
// its children are all pruned, we may prune it too.
// if (childrenPruned && node.NumChildren() > 0)
// {
if (childrenPruned && node.NumChildren() > 0)
{
// Log::Warn << "Prune parent node " << node.Point(0) << "c" <<
//node.NumDescendants() << ".\n";
// node.Stat().Pruned() = true;
// }
node.Stat().Pruned() = true;
node.Stat().Owner() = centroids.n_cols;
}
// if (node.NumChildren() > 0)
// if (node.Child(0).Stat().Pruned() && !node.Child(1).Stat().Pruned())
// Log::Warn << "Node left child pruned but right child not:\n" <<