The AllCategoricalSplit is a splitting function that will split categorical features into many children: one child for each category. More...
Classes | |
class | AuxiliarySplitInfo |
Static Public Member Functions | |
template < typename ElemType > | |
static size_t | CalculateDirection (const ElemType &point, const arma::Col< ElemType > &classProbabilities, const AuxiliarySplitInfo< ElemType > &) |
Calculate the direction a point should percolate to. More... | |
template < typename ElemType > | |
static size_t | NumChildren (const arma::Col< ElemType > &classProbabilities, const AuxiliarySplitInfo< ElemType > &) |
Return the number of children in the split. More... | |
template<bool UseWeights, typename VecType , typename WeightVecType > | |
static double | SplitIfBetter (const double bestGain, const VecType &data, const size_t numCategories, const arma::Row< size_t > &labels, const size_t numClasses, const WeightVecType &weights, const size_t minimumLeafSize, const double minimumGainSplit, arma::Col< typename VecType::elem_type > &classProbabilities, AuxiliarySplitInfo< typename VecType::elem_type > &aux) |
Check if we can split a node. More... | |
The AllCategoricalSplit is a splitting function that will split categorical features into many children: one child for each category.
FitnessFunction | Fitness function to evaluate gain with. |
Definition at line 28 of file all_categorical_split.hpp.
|
static |
Calculate the direction a point should percolate to.
classProbabilities | Auxiliary information for the split. |
aux | (Unused) auxiliary information for the split. |
|
static |
Return the number of children in the split.
classProbabilities | Auxiliary information for the split. |
aux | (Unused) auxiliary information for the split. |
|
static |
Check if we can split a node.
If we can split a node in a way that improves on 'bestGain', then we return the improved gain. Otherwise we return the value 'bestGain'. If a split is made, then classProbabilities and aux may be modified. For this particular split type, aux will be empty and classProbabilities will hold one element—the number of children.
bestGain | Best gain seen so far (we'll only split if we find gain better than this). |
data | The dimension of data points to check for a split in. |
numCategories | Number of categories in the categorical data. |
labels | Labels for each point. |
numClasses | Number of classes in the dataset. |
minimumLeafSize | Minimum number of points in a leaf node for splitting. |
classProbabilities | Class probabilities vector, which may be filled with split information a successful split. |
aux | Auxiliary split information, which may be modified on a successful split. |