Esempio n. 1
0
File: CAE1.0.py Progetto: Lu-Nie/ML
# 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