Skip to content

glassbox.models.trees.classifier

DecisionTreeClassifier — splits on Gini impurity.

DecisionTreeClassifier 

DecisionTreeClassifier(max_depth=100, min_samples_split=2)

Bases: BaseTree

A decision tree classifier.

Source code in glassbox/models/trees/_base.py 
85
86
87
88
89
90
91
92
93
94
95
96
97
98
def __init__(self, max_depth: int = 100, min_samples_split: int = 2) -> None:
    """
    Initialize the base tree model.

    Parameters
    ----------
    max_depth : int, default=100
        Maximum depth of the tree.
    min_samples_split : int, default=2
        Minimum number of samples required to split an internal node.
    """
    self.max_depth = max_depth if max_depth is not None else float("inf")
    self.min_samples_split = min_samples_split
    self.root: Optional[_Node] = None