Newer
Older
import numpy as np
from supersvd import supersvd
import matplotlib.pyplot as plt
ts = np.fromfile('ts.std', dtype=np.float32).reshape(1147, 28, 31)
ps = np.fromfile('ps.std', dtype=np.float32).reshape(1147, 28, 31)
svd = supersvd(ts, ps, 4)
first_mode_ts = svd.x_vect[0]
plt.matshow(first_mode_ts)
plt.savefig('first_mode_ts.png')
first_ps_timeseries = svd.x_coeff[0]
plt.figure()
plt.plot(first_ps_timeseries)
plt.savefig('first_ps_timeseries.png')
print('Correlation coefficient between first_ts_timeseries. and first_ps_timeseries:', svd.corrcoeff[0])