def test_petrosian_fd(self):
pfd = petrosian_fd(RANDOM_TS)
petrosian_fd(list(RANDOM_TS))
self.assertEqual(np.round(pfd, 3), 1.030)
# 2D data
assert_equal(aal(petrosian_fd, axis=1, arr=data), petrosian_fd(data))
assert_equal(aal(petrosian_fd, axis=0, arr=data),
petrosian_fd(data, axis=0))
def get_PFD_at(self, channel):
"""
Get Petrosial Fractal Dimension.
:param channel: Channel to compute Petrosian fractal.
:return: Petrosian fractal dimension.
"""
channel_data = self.__get_raw_data_at__(channel)
pfd = entropy.petrosian_fd(channel_data)
return pfd
def PetrosianFract(RR_windows, **kwargs):
feat = []
for wRR in RR_windows:
try:
value = entropy.petrosian_fd(wRR)
except:
value = np.nan
feat.append(value)
return feat
def petrosian(x):
return entropy.petrosian_fd(x)
def test_petrosian_fd(self):
pfd = petrosian_fd(RANDOM_TS)
petrosian_fd(list(RANDOM_TS))
self.assertEqual(np.round(pfd, 3), 1.030)