|
a |
|
b/MainDecExample.py |
|
|
1 |
""" |
|
|
2 |
|
|
|
3 |
Stefania Fresca, MOX Laboratory, Politecnico di Milano |
|
|
4 |
April 2019 |
|
|
5 |
|
|
|
6 |
""" |
|
|
7 |
|
|
|
8 |
import os |
|
|
9 |
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' |
|
|
10 |
import sys |
|
|
11 |
sys.stdout = open('*.out', 'w') |
|
|
12 |
|
|
|
13 |
import utils |
|
|
14 |
from DecNet import DecNet |
|
|
15 |
|
|
|
16 |
if __name__ == '__main__': |
|
|
17 |
config = dict() |
|
|
18 |
config['n'] = |
|
|
19 |
config['n_params'] = |
|
|
20 |
config['lr'] = |
|
|
21 |
config['omega_h'] = |
|
|
22 |
config['omega_n'] = |
|
|
23 |
config['batch_size'] = |
|
|
24 |
config['n_data'] = |
|
|
25 |
config['N_h'] = |
|
|
26 |
config['n_h'] = |
|
|
27 |
config['N_t'] = |
|
|
28 |
config['train_mat'] = '' |
|
|
29 |
config['test_mat'] = '' |
|
|
30 |
config['train_params'] = '' |
|
|
31 |
config['test_params'] = '' |
|
|
32 |
config['checkpoints_folder'] = '' |
|
|
33 |
config['graph_folder'] = '' |
|
|
34 |
config['large'] = |
|
|
35 |
config['zero_padding'] = |
|
|
36 |
config['p'] = |
|
|
37 |
config['restart'] = |
|
|
38 |
|
|
|
39 |
model = DecNet(config) |
|
|
40 |
model.build() |
|
|
41 |
model.test_all() |