import numpy as np
from pyhht import EMD
import matplotlib.pyplot as plt
from pyhht.visualization import plot_imfs
from time import time
import lee_dataset_CWRU
from lee_dataset_CWRU import *
DE_H1, FE_H1, t_DE_H1, t_FE_H1, RPM_H1, samples_s_DE_H1, samples_s_FE_H1 = lee_dataset('97.mat')
x = DE_H1
t = t_DE_H1
plt.figure(figsize=(7, 4))
plt.plot(t, x)
time0 = time()
decomposer = EMD(x)
imfs = decomposer.decompose()
elapsed_time = time() - time0
Mode 2, Iteration 600
2 minima > 0; 19594 maxima < 0.
Mode 2, Iteration 800
1 minima > 0; 19640 maxima < 0.
Mode 2, Iteration 1000
1 minima > 0; 19677 maxima < 0.
Mode 4, Iteration 1200
3 minima > 0; 7410 maxima < 0.
Mode 5, Iteration 1400
2 minima > 0; 4185 maxima < 0.
Mode 6, Iteration