# In[5]: import tensorflow # In[6]: import tensorflow # In[ ]: kinds = [ 'boss_cv', 'boss_da+ms', 'boss_db', 'boss_db+ms', 'boss_dq', 'boss_dz', 'fgkm', 'hotstars', 'wd', 'wdsb2', 'yso', 'hotstars_m' ] flux_cv, spectrum_cv = DP.Preprocessing7('/home/njl/ML/optical/' + kinds[0] + '/' + '*.fit') flux_dams, spectrum_dams = DP.Preprocessing7('/home/njl/ML/optical/' + kinds[1] + '/' + '*.fit') flux_db, spectrum_db = DP.Preprocessing7('/home/njl/ML/optical/' + kinds[2] + '/' + '*.fit') flux_db.pop(324) spectrum_db.pop(324) # In[115]: size = (2, 4096) X_train = spectrum_cv + spectrum_dams + spectrum_db X_train = np.stack(X_train) X_train = X_train.reshape(size[0], len(X_train), size[1]) #change the shape to NHWC for CAE input