policy.py 文件源码

def create_network(**kwargs):
        defaults = {"timesteps": 128, "data_dim": 15}
        params = defaults
        params.update(**kwargs)

        network = Sequential()
        network.add(LSTM(output_dim=16,
                         activation='sigmoid',
                         inner_activation='hard_sigmoid',
                         input_shape=(params['timesteps'], params['data_dim']
                                      )))
        network.add(Dropout(0.15))
        network.add(Dense(1))
        # network.add(LeakyReLU(alpha=0.5))
        network.add(Activation('relu'))

        network.compile(optimizer='rmsprop',
                        loss='binary_crossentropy',
                        metrics=['accuracy'])
        return network