Python Autocorrelation.fit_transform示例

示例#1

 def test_fit_transform(self):
     a = Autocorrelation(depths=[1], properties=['I'])
     expected = numpy.array([
         [8],
         [8],
         [104],
     ])
     self.assertTrue((a.fit_transform(ALL_DATA) == expected).all())

示例#2

文件： test_molecule.py 项目： aced125/molml

 def test_get_labels(self):
     a = Autocorrelation(depths=[1, 2], properties=['I', 'EN'])
     X = a.fit_transform([METHANE])
     labels = a.get_labels()
     self.assertEqual(X.shape[1], len(labels))
     expected = (
         'EN_1', 'EN_2',
         'I_1', 'I_2',
     )
     self.assertEqual(labels, expected)