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))
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))