from .utils import Bunch

data_params = {'T': 100., 'length_scale': 5., 'N': 20}

model_params = {'hidden_units': 8,
                'noise_size': 8,
                'batch_size': 30,
                'num_epochs': 50,
                'num_steps': 15,
                'compensator_type': "analytical",
                'save_every': 500,
                'learning_rate': 0.001,
                'init_value': .0001,
                'log_dir': 'saved_models',
                'store_dir': 'saved_models',
                'debug': False,
                'log': "logging level",
                'numpy_seed': None,
                'tf_seed': None,
                'log': "INFO",
                'int_poisson': 'gauss',
                'autoencode': False,
                # 'esn': False
                }

# transform the model parameters to a namespace required by the model


args = Bunch(model_params)
