from models import *
import numpy as np
from keras.callbacks import ModelCheckpoint, CSVLogger, EarlyStopping
from keras.models import load_model
import h5py
from seq_exp import SeqExp


if __name__ == "__main__":
        '''
        These are the training setting. 
        '''
        dset = 'UR-Filled'
        img_width, img_height, win_len, epochs = 64,64, 8,50
        stride = 1
       
        model, model_name, model_type = CLSTM_AE(img_width, img_height, win_len)
       # model, model_name, model_type = dummy_3d(img_width, img_height, win_len)

        print('model loaded')
        print(model.summary())

        exp_3D = SeqExp(model = model, model_name = model_name, epochs = epochs, \
                win_len = win_len, dset = dset, img_width = img_width, img_height = img_height)
        exp_3D.set_train_data()

        print(exp_3D.train_data.shape)
        print('data loaded')
        exp_3D.train()