Source code for rocelib.models.sklearn_models.TrainableDecisionTreeModel
from sklearn.tree import DecisionTreeClassifier
from rocelib.models.sklearn_models.TrainableSKLearnModel import TrainableSKLearnModel
[docs]
class TrainableDecisionTreeModel(TrainableSKLearnModel):
"""
A Decision Tree Classifier model wrapper for scikit-learn.
Inherits from TrainableSKLearnModel and initializes a DecisionTreeClassifier as the underlying model.
"""
def __init__(self):
super().__init__(DecisionTreeClassifier())