def test_correlation_fit_trans(self): data, label = get_data_label(load_boston()) method = SelectionMethod.Correlation(0.60) selector = Selective(method) subset = selector.fit_transform(data) self.assertListEqual(list(subset.columns), ['CRIM', 'ZN', 'INDUS', 'CHAS', 'RM', 'PTRATIO', 'B'])
def test_variance_lt1_fit_trans(self): data, label = get_data_label(load_boston()) method = SelectionMethod.Variance(threshold=1.0) selector = Selective(method) subset = selector.fit_transform(data) # Reduced columns self.assertEqual(data.shape[1], 13) self.assertEqual(subset.shape[1], 10) self.assertListEqual(list(subset.columns), [ 'CRIM', 'ZN', 'INDUS', 'AGE', 'DIS', 'RAD', 'TAX', 'PTRATIO', 'B', 'LSTAT' ])