예제 #1
0
 def test_with_defaults(self):
     trainable = DecisionTreeClassifier()
     trained = trainable.fit(self.train_X, self.train_y)
     _ = trained.predict(self.test_X)
예제 #2
0
 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)