def test_perm_entropy(self):
self.assertEqual(
np.round(perm_entropy(RANDOM_TS, order=3, delay=1, normalize=True),
1), 1.0)
# Compare with Bandt and Pompe 2002
self.assertEqual(np.round(perm_entropy(BANDT_PERM, order=2), 3), 0.918)
self.assertEqual(np.round(perm_entropy(BANDT_PERM, order=3), 3), 1.522)
# Error
with self.assertRaises(ValueError):
perm_entropy(BANDT_PERM, order=4, delay=3)
with self.assertRaises(ValueError):
perm_entropy(BANDT_PERM, order=3, delay=0.5)
with self.assertRaises(ValueError):
perm_entropy(BANDT_PERM, order=1, delay=1)
def PermEn(RR_windows, **kwargs):
feat = []
for wRR in RR_windows:
try:
value = entropy.perm_entropy(wRR, order=3, normalize=True)
except:
value = np.nan
feat.append(value)
return feat
def etrpy(sample, etype):
if etype == "svd":
et = entropy.svd_entropy(sample, order=3, delay=1)
elif etype == "spectral":
et = entropy.spectral_entropy(sample,
100,
method='welch',
normalize=True)
elif etype == "sample":
et = entropy.sample_entropy(sample, order=3)
elif etype == "perm":
et = entropy.perm_entropy(sample, order=3, normalize=True)
else:
print("Error: unrecognised entropy type {}".format(etype))
exit(-1)
return et
def createEntropyFeatureArray(self, epochSeries : pd.Series, samplingFreq : int) -> (np.ndarray, List[str]):
''' Creates 3d Numpy with a entropy features - also returns the feature names
Creates the following features:
- Approximate Entropy (AE)
- Sample Entropy (SamE)
- Spectral Entropy (SpeE)
- Permutation Entropy (PE)
- Singular Value Decomposition Entropy (SvdE)
For each channel there are 5 features then
NaN Values will be set to Zero (not good but it works for now)
'''
# Create np array, where the data will be stored
d1 = len(epochSeries) # First Dimesion
d2 = 1 # only one sample in that epoch
channels = len(epochSeries[0].columns)
d3 = channels * 5 # second dimension - 5 because we calculate five different entropies for each channel
entropyFeatureArrayX = createEmptyNumpyArray(d1, d2, d3)
# Create a list where all feature names are stored
entropyFeatureList = [None] * d3
stepSize = 5 # step is 5 because we calculate 5 different entropies
for i in range (0, len(epochSeries)): # loop through the epochs
# We start the the stepz size and loop through the columns, but we have to multiply by the stepzsize and add once the step size (because we don't start at 0)
for j in range(stepSize, (len(epochSeries[i].columns)*stepSize)+stepSize, stepSize): # loop through the columns
# j_epoch is the normalized index for the epoch series (like the step size would be 1)
j_epoch = j/stepSize - 1
# get the column name
col = epochSeries[i].columns[j_epoch]
# The values of the epoch of the current column
colEpochList = epochSeries[i][col].tolist()
######################################
# calculate Approximate Entropy
# ------------------------------------
val = entropy.app_entropy(colEpochList, order=2)
# if the value is NaN, just set it to 0
if np.isnan(val):
val = 0
entropyFeatureArrayX[i][0][j-1] = val
# add approximate entropy feature to the list
entropyFeatureList = addFeatureToList(featureList = entropyFeatureList,
featureListIndex = j-1,
newFeatureName = "{col}_approximate_entropy".format(col=col))
######################################
# calculate Sample Entropy
# ------------------------------------
val = entropy.sample_entropy(colEpochList, order=2)
# if the value is NaN, just set it to 0
if np.isnan(val):
val = 0
entropyFeatureArrayX[i][0][j-2] = val
entropyFeatureList = addFeatureToList(featureList = entropyFeatureList,
featureListIndex = j-2,
newFeatureName = "{col}_sample_entropy".format(col=col))
######################################
# calculate Spectral Entropy
# ------------------------------------
val = entropy.spectral_entropy(colEpochList, sf=samplingFreq ,method='fft', normalize=True)
# if the value is NaN, just set it to 0
if np.isnan(val):
val = 0
entropyFeatureArrayX[i][0][j-3] = val
entropyFeatureList = addFeatureToList(featureList = entropyFeatureList,
featureListIndex = j-3,
newFeatureName = "{col}_spectral_entropy".format(col=col))
######################################
# calculate Permutation Entropy
# ------------------------------------
val = entropy.perm_entropy(colEpochList, order=3, normalize=True, delay=1)
# if the value is NaN, just set it to 0
if np.isnan(val):
val = 0
entropyFeatureArrayX[i][0][j-4] = val
entropyFeatureList = addFeatureToList(featureList = entropyFeatureList,
featureListIndex = j-4,
newFeatureName = "{col}_permutation_entropy".format(col=col))
######################################
# calculate Singular Value Decomposition entropy.
# ------------------------------------
val = entropy.svd_entropy(colEpochList, order=3, normalize=True, delay=1)
# if the value is NaN, just set it to 0
if np.isnan(val):
val = 0
entropyFeatureArrayX[i][0][j-5] = val
entropyFeatureList = addFeatureToList(featureList = entropyFeatureList,
featureListIndex = j-5,
newFeatureName = "{col}_svd_entropy".format(col=col))
#break
#break
# Normalize everything to 0-1
print("Normalizing the entropy features...")
# Norm=max -> then it will normalize between 0-1, axis=0 is important too!
# We need to reshape it to a 2d Array
X_entropy_norm = preprocessing.normalize(entropyFeatureArrayX.reshape(entropyFeatureArrayX.shape[0], entropyFeatureArrayX.shape[2]), norm='max', axis=0)
# Now reshape it back to a simple 3D array
X_entropy_norm = X_entropy_norm.reshape(X_entropy_norm.shape[0], 1, X_entropy_norm.shape[1])
return X_entropy_norm, entropyFeatureList
def perm_entropy(x):
return entropy.perm_entropy(x, order=3, normalize=True)
# color = 'tab:red'
# ax1.set_xlabel('iter')
# ax1.set_ylabel('X', color=color)
# ax1.plot(s[si], color=color)
# ax1.tick_params(axis='y', labelcolor=color)
#
# ax2 = ax1.twinx()
# color = 'tab:blue'
# ax2.set_ylabel('S', color=color)
# ax2.plot(S[si], color=color)
# ax2.tick_params(axis='y', labelcolor=color)
#
# plt.show()
# Entropy:
print(entropy.perm_entropy(s[0], order=3,
normalize=True)) # Permutation entropy
print(entropy.spectral_entropy(s[0], 100, method='welch',
normalize=True)) # Spectral entropy
print(entropy.svd_entropy(
s[0], order=3, delay=1,
normalize=True)) # Singular value decomposition entropy
print(entropy.app_entropy(s[0], order=2,
metric='chebyshev')) # Approximate entropy
print(entropy.sample_entropy(s[0], order=2,
metric='chebyshev')) # Sample entropy
fpath_db = os.path.join(os.path.dirname(__file__), 'data',
'06-sir-gamma-beta.sqlite3')
te = TrajectoryEnsemble(fpath_db).stats()
s = te.traj[1].get_signal().series
print(entropy.app_entropy(s[0], order=2,
