-
Notifications
You must be signed in to change notification settings - Fork 0
/
hyper_parameter.py
34 lines (28 loc) · 982 Bytes
/
hyper_parameter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# -*- coding: utf-8 -*-
from __future__ import division, print_function, absolute_import
class HyperParameter(object):
def __init__(self):
# rnn parameters
self.rnn_batch_size = 32
self.rnn_input_steps = 20
self.rnn_predict_steps = 1
self.rnn_input_depth = 1
self.rnn_predict_depth = 1
self.rnn_hidden = 20
self.rnn_train_epoch = 2
self.rnn_model_dir = 'model'
self.encoder_dropout = 1.0
self.encoder_stability_loss = 0.0
self.encoder_activation_loss = 1e-05
self.decoder_stability_loss = 0.0
self.decoder_activation_loss = 1e-05
self.encoder_rnn_layers = 1
self.decoder_rnn_layers = 1
# ar parameters
self.ar_batch_size = 128
self.ar_input_steps = 30
self.ar_predict_steps = 1
self.ar_input_depth = 1
self.ar_predict_depth = 1
self.ar_train_epoch = 2
self.ar_model_dir = 'model'