13 #ifndef MLPACK_METHODS_HOEFFDING_TREES_HOEFFDING_TREE_HPP 14 #define MLPACK_METHODS_HOEFFDING_TREES_HOEFFDING_TREE_HPP 55 template<
typename FitnessFunction = GiniImpurity,
56 template<
typename>
class NumericSplitType =
58 template<
typename>
class CategoricalSplitType =
59 HoeffdingCategoricalSplit
91 template<
typename MatType>
94 const arma::Row<size_t>& labels,
95 const size_t numClasses,
96 const bool batchTraining =
true,
97 const double successProbability = 0.95,
98 const size_t maxSamples = 0,
99 const size_t checkInterval = 100,
100 const size_t minSamples = 100,
101 const CategoricalSplitType<FitnessFunction>& categoricalSplitIn
102 = CategoricalSplitType<FitnessFunction>(0, 0),
103 const NumericSplitType<FitnessFunction>& numericSplitIn =
104 NumericSplitType<FitnessFunction>(0));
126 const size_t numClasses,
127 const double successProbability = 0.95,
128 const size_t maxSamples = 0,
129 const size_t checkInterval = 100,
130 const size_t minSamples = 100,
131 const CategoricalSplitType<FitnessFunction>& categoricalSplitIn
132 = CategoricalSplitType<FitnessFunction>(0, 0),
133 const NumericSplitType<FitnessFunction>& numericSplitIn =
134 NumericSplitType<FitnessFunction>(0),
135 std::unordered_map<
size_t, std::pair<size_t, size_t>>*
136 dimensionMappings = NULL);
165 template<
typename MatType>
166 void Train(
const MatType& data,
167 const arma::Row<size_t>& labels,
168 const bool batchTraining =
true);
174 template<
typename MatType>
175 void Train(
const MatType& data,
177 const arma::Row<size_t>& labels,
178 const bool batchTraining =
true);
186 template<
typename VecType>
187 void Train(
const VecType& point,
const size_t label);
244 template<
typename VecType>
254 template<
typename VecType>
255 size_t Classify(
const VecType& point)
const;
271 template<
typename VecType>
272 void Classify(
const VecType& point,
size_t& prediction,
double& probability)
282 template<
typename MatType>
283 void Classify(
const MatType& data, arma::Row<size_t>& predictions)
const;
296 template<
typename MatType>
298 arma::Row<size_t>& predictions,
299 arma::rowvec& probabilities)
const;
307 template<
typename Archive>
308 void serialize(Archive& ar,
const unsigned int );
314 std::vector<NumericSplitType<FitnessFunction>> numericSplits;
316 std::vector<CategoricalSplitType<FitnessFunction>> categoricalSplits;
319 std::unordered_map<size_t, std::pair<size_t, size_t>>* dimensionMappings;
330 size_t checkInterval;
338 double successProbability;
343 size_t splitDimension;
345 size_t majorityClass;
348 double majorityProbability;
350 typename CategoricalSplitType<FitnessFunction>::SplitInfo categoricalSplit;
352 typename NumericSplitType<FitnessFunction>::SplitInfo numericSplit;
354 std::vector<HoeffdingTree*> children;
360 #include "hoeffding_tree_impl.hpp" HoeffdingTree()
Construct a Hoeffding tree with no data and no information.
const HoeffdingTree & Child(const size_t i) const
Get a child.
size_t NumChildren() const
Get the number of children.
Auxiliary information for a dataset, including mappings to/from strings (or other types) and the data...
~HoeffdingTree()
Clean up memory.
The HoeffdingTree object represents all of the necessary information for a Hoeffding-bound-based deci...
size_t CheckInterval() const
Get the number of samples before a split check is performed.
The core includes that mlpack expects; standard C++ includes and Armadillo.
void serialize(Archive &ar, const unsigned int)
Serialize the split.
void Train(const MatType &data, const arma::Row< size_t > &labels, const bool batchTraining=true)
Train on a set of points, either in streaming mode or in batch mode, with the given labels...
size_t CalculateDirection(const VecType &point) const
Given a point and that this node is not a leaf, calculate the index of the child node this point woul...
size_t Classify(const VecType &point) const
Classify the given point, using this node and the entire (sub)tree beneath it.
size_t MaxSamples() const
Get the maximum number of samples before a split is forced.
double SuccessProbability() const
Get the confidence required for a split.
void CreateChildren()
Given that this node should split, create the children.
CategoricalSplitType< FitnessFunction > CategoricalSplit
Allow access to the categorical split type.
double MajorityProbability() const
Get the probability of the majority class (based on training samples).
size_t NumDescendants() const
Get the size of the Hoeffding Tree.
HoeffdingNumericSplit< FitnessFunction, double > HoeffdingDoubleNumericSplit
Convenience typedef.
size_t & MajorityClass()
Modify the majority class.
size_t SplitDimension() const
Get the splitting dimension (size_t(-1) if no split).
size_t MinSamples() const
Get the minimum number of samples for a split.
size_t SplitCheck()
Check if a split would satisfy the conditions of the Hoeffding bound with the node's specified succes...
HoeffdingTree & Child(const size_t i)
Modify a child.
size_t MajorityClass() const
Get the majority class.
NumericSplitType< FitnessFunction > NumericSplit
Allow access to the numeric split type.
double & MajorityProbability()
Modify the probability of the majority class.