示例#1
0
 def test_fixed_init_rf(self):
     model = BaseModel('c-rf', 'test_params_rf_set')
     model.fit(X, y)
     self.assertEqual(model.predict_proba(X).shape[0], n_samples)
     self.assertEqual(model.predict_proba(X).shape[1], num_classes)