mlpack  master
dt_utils.hpp File Reference
Include dependency graph for dt_utils.hpp:

Go to the source code of this file.

Classes

class  PathCacher
 This class is responsible for caching the path to each node of the tree. More...

 

Namespaces

 mlpack
 
Linear algebra utility functions, generally performed on matrices or vectors.
 
 mlpack::det
 
Density Estimation Trees.
 

Functions

template
<
typename
MatType
,
typename
TagType
>
void PrintLeafMembership (DTree< MatType, TagType > *dtree, const MatType &data, const arma::Mat< size_t > &labels, const size_t numClasses, const std::string &leafClassMembershipFile="")
 Print the membership of leaves of a density estimation tree given the labels and number of classes. More...

 
template
<
typename
MatType
,
typename
TagType
>
void PrintVariableImportance (const DTree< MatType, TagType > *dtree, const std::string viFile="")
 Print the variable importance of each dimension of a density estimation tree. More...

 
template
<
typename
MatType
,
typename
TagType
>
DTree< MatType, TagType > * Trainer (MatType &dataset, const size_t folds, const bool useVolumeReg=false, const size_t maxLeafSize=10, const size_t minLeafSize=5, const std::string unprunedTreeOutput="", const bool skipPruning=false)
 Train the optimal decision tree using cross-validation with the given number of folds. More...

 

Detailed Description

Author
Parikshit Ram (pram@.nosp@m.cc.g.nosp@m.atech.nosp@m..edu)

This file implements functions to perform different tasks with the Density Tree class.

mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.

Definition in file dt_utils.hpp.