@@ -38,35 +38,32 @@ def get_hyperparam(self, param_name: str, hp=None) -> Any:
3838 def build (self , hp : Optional [Any ]) -> Model :
3939 """Get keras model with string and feature input and single binary out"""
4040 line_lstm_dropout_rate = self .get_hyperparam ("line_lstm_dropout_rate" , hp )
41- line_lstm_recurrent_dropout_rate = self .get_hyperparam ("line_lstm_recurrent_dropout_rate" , hp )
4241 variable_lstm_dropout_rate = self .get_hyperparam ("variable_lstm_dropout_rate" , hp )
43- variable_lstm_recurrent_dropout_rate = self .get_hyperparam ("variable_lstm_recurrent_dropout_rate" , hp )
4442 value_lstm_dropout_rate = self .get_hyperparam ("value_lstm_dropout_rate" , hp )
45- value_lstm_recurrent_dropout_rate = self .get_hyperparam ("value_lstm_recurrent_dropout_rate" , hp )
4643 dense_a_drop = self .get_hyperparam ("dense_a_drop" , hp )
4744 dense_b_drop = self .get_hyperparam ("dense_b_drop" , hp )
4845
4946 line_input = Input (shape = (None , self .line_shape [2 ]), name = "line_input" , dtype = self .d_type )
5047 line_lstm = LSTM (units = self .line_shape [1 ],
5148 dtype = self .d_type ,
5249 dropout = line_lstm_dropout_rate ,
53- recurrent_dropout = line_lstm_recurrent_dropout_rate )
50+ recurrent_dropout = 0 )
5451 line_bidirectional = Bidirectional (layer = line_lstm , name = "line_bidirectional" )
5552 line_lstm_branch = line_bidirectional (line_input )
5653
5754 variable_input = Input (shape = (None , self .variable_shape [2 ]), name = "variable_input" , dtype = self .d_type )
5855 variable_lstm = LSTM (units = self .variable_shape [1 ],
5956 dtype = self .d_type ,
6057 dropout = variable_lstm_dropout_rate ,
61- recurrent_dropout = variable_lstm_recurrent_dropout_rate )
58+ recurrent_dropout = 0 )
6259 variable_bidirectional = Bidirectional (layer = variable_lstm , name = "variable_bidirectional" )
6360 variable_lstm_branch = variable_bidirectional (variable_input )
6461
6562 value_input = Input (shape = (None , self .value_shape [2 ]), name = "value_input" , dtype = self .d_type )
6663 value_lstm = LSTM (units = self .value_shape [1 ],
6764 dtype = self .d_type ,
6865 dropout = value_lstm_dropout_rate ,
69- recurrent_dropout = value_lstm_recurrent_dropout_rate )
66+ recurrent_dropout = 0 )
7067 value_bidirectional = Bidirectional (layer = value_lstm , name = "value_bidirectional" )
7168 value_lstm_branch = value_bidirectional (value_input )
7269
0 commit comments