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, util::Timers &timers=IO::GetTimers()) |
| Train the optimal decision tree using cross-validation with the given number of folds. More... | |
Detailed Description
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.