def test_with_defaults(self): trainable = DecisionTreeClassifier() trained = trainable.fit(self.train_X, self.train_y) _ = trained.predict(self.test_X)
def test_ccp_alpha(self): trainable = DecisionTreeClassifier(ccp_alpha=0.01) trained = trainable.fit(self.train_X, self.train_y) predicted = trained.predict(self.test_X)