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