Add ClassProbabilities() member to DecisionTree.

This commit is contained in:
Ryan Curtin
2023-07-07 11:04:02 -04:00
parent 9b89f79e01
commit a7c5a3ba23
@@ -470,6 +470,11 @@ class DecisionTree :
//! trained tree).
size_t SplitDimension() const { return splitDimension; }
//! Get the class probabilities, if this is a leaf node in the trained tree.
//! Note that if this is not a leaf, then this may contain arbitrary
//! information used by the split in the tree!
const arma::vec& ClassProbabilities() const { return classProbabilities; }
/**
* Given a point and that this node is not a leaf, calculate the index of the
* child node this point would go towards. This method is primarily used by