将 verbose 设置为 0 不会按预期工作

Question

我开发了一个深度 Q 学习代理。 我在model.fit() function 中使用了verbose=0并且仍然得到以下行 output：

1/1 [==============================] - 0s 12ms/step

这是我构建 model 的方法：

def build_model(self):
    # Build the model
    model = Sequential()
    model.add(Dense(24, input_dim=STATE_SIZE, activation='relu'))
    model.add(Dense(24, activation='relu'))
    model.add(Dense(ACTION_SIZE, activation='linear'))
    model.compile(loss='mse', optimizer=Adam(lr=self.learning_rate))
    #if os.path.isfile(self.weight_backup):
    #    model.load_weights(self.weight_backup)
    #    self.epsilon = self.epsilon_min
    return model

我在 replay() function 中拟合数据：

def replay(self, batch_size):
    # training
    if len(self.memory) < batch_size:
        return
    mini_batch = random.sample(self.memory, batch_size)
    for state, action, reward, next_state, done in mini_batch:
        if done:
            target = reward
        else:
            target = reward + self.gamma * np.amax(self.model.predict(next_state)[0])
        train_target = self.model.predict(state)
        train_target[0][action] = target
        #callbacks = [ProgbarLogger()]
        self.model.fit(state, train_target, epochs=1, verbose=0)

有没有办法防止output？

Answer 1

我找到了解决方案。 它来自predict() function 并且在将 verbose 设置为 0 后修复了问题