test_recurrent_stress_tests.py 文件源码

def test_many_to_many(self):
        params = dict(
            input_dims=[1, 10, 5], go_backwards=False, activation='tanh',  # fails with hard_sigmoid
            stateful=False, unroll=False, return_sequences=True, output_dim=1
        ),
        model = Sequential()
        model.add(LSTM(output_dim=params[0]['output_dim'],
                       input_shape=(10, 5),
                       activation=params[0]['activation'],
                       inner_activation='sigmoid',
                       return_sequences=True,
                       ))
        relative_error, keras_preds, coreml_preds = simple_model_eval(params, model)
        # print relative_error, '\n', keras_preds, '\n', coreml_preds, '\n'
        for i in range(len(relative_error)):
            self.assertLessEqual(relative_error[i], 0.01)