Python ColsExtractor示例

示例#1

文件： feature_extractors_test.py 项目： josericardo/mlpal

    def test_extracts_more_than_one_col(self):
        ce = ColsExtractor([0, 2])
        features = ce.transform(self.X)

        expected = [
            [0, 2],
            [3, 5]
        ]

        self.assertTrue(np.array_equal(expected, features))

示例#2

文件： feature_extractors_test.py 项目： josericardo/mlpal

 def test_extracts_one_col(self):
     ce = ColsExtractor([1])
     self.assertEqual([[1], [4]], ce.transform(self.X))

示例#3

文件： feature_extractors_test.py 项目： eliasdorneles/mlpal

    def test_extracts_and_flattens(self):
        ce = ColsExtractor([0], flat=True)
        features = ce.transform(self.X)

        expected = [0, 3]
        self.assertTrue(np.array_equal(expected, features))