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

Go to the source code of this file.

Namespaces

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

Functions

void PrintLeafMembership (DTree *dtree, const arma::mat &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...

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

 
DTree * Trainer (arma::mat &dataset, const size_t folds, const bool useVolumeReg=false, const size_t maxLeafSize=10, const size_t minLeafSize=5, const std::string unprunedTreeOutput="")
 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.