fmda/params.yaml

   1 # File used to store hyperparameters.
   2
   3 # Stateless RNN, batch_size declared at fit step
   4 rnn_stateless:
   5   batch_size: 32
   6   optimizer: adam
   7   epochs: 100
   8   rnn_units: 6
   9   activation: ['linear', 'linear']
  10   dropout: 0.2
  11   recurrent_dropout: 0.2
  12   learning_rate: 0.001
  13
  14 # Stateful RNN, batch_input_shape used
  15 rnn_stateful:
  16   batch_size: 32
  17   optimizer: adam
  18   epochs: 100
  19   rnn_units: 6
  20   activation: ['linear', 'linear']
  21   dropout: 0.2
  22   recurrent_dropout: 0.2
  23   learning_rate: 0.001
  24
  25
  26
  27
  28
  29
  30