Add this logic to SD4 as a member function. Consider adding a .fit() interface.
coords_all = np.reshape(coords, (len(coords), coords.shape[1] * coords.shape[2])).T
avg_coords_all = np.mean(coords_all, 1)
tmp_all = np.reshape(
np.tile(avg_coords_all, coords.shape[0]),
(coords.shape[0], coords.shape[1] * coords.shape[2]),
).T
devs_md_all = coords_all - tmp_all
ZPrj4 = W.dot(devs_md_all)
Add this logic to SD4 as a member function. Consider adding a .fit() interface.