Ejemplo n.º 1
0
 def test_extract_important_features_2d_array(self):
     importances = _extract_important_features(
         self.features.ids('observation'),
         np.ndarray((2, 3), buffer=np.array([0.1, 0.2, 0.3, 0.4, 0.5,
                                             0.6])))
     self.assertEqual(sorted(self.exp_svm), sorted(importances))
Ejemplo n.º 2
0
 def test_extract_important_features_1d_array(self):
     importances = _extract_important_features(
         self.features.ids('observation'),
         np.ndarray((3, ), buffer=np.array([0.1, 0.2, 0.3])))
     self.assertEqual(sorted(self.exp_rf), sorted(importances))