python - Keras属性错误: 'History' object has no attribute 'predict'

Question

Note: I have seen this related post but I don't know I can use the answer for my problem.

我尝试使用 Keras 进行简单回归。为此，我创建了一个简单的 policy_network() 函数，它返回模型。

def policy_network():
    model = Sequential()
    model.add(MaxPooling2D(pool_size=(4, 4),input_shape=[64,64,3]))
    model.add(Flatten())
    model.add(Dense(1, kernel_initializer='normal', activation='linear'))

    model.compile(loss='mean_squared_error',
                  optimizer=Adam(lr=learning_rate),
                  metrics=['mean_squared_error'])

    return model

我还定义了一个全局变量policy_network。我使用以下作业

policy_network = policy_network().fit(images, actions,
                  batch_size=256,
                  epochs=10,
                  shuffle=True)

但是当我打电话

action = policy_network.predict(image)

我收到AttributeError:'History'对象没有属性'predict'

Answer 1

Keras 的 fit() 确实不返回模型，但它返回一个包含每个周期损失和指标的 History 对象。您正在使用的代码模式根本不适用于 Keras。

这样做:

model = policy_network()
model.fit(images, actions,
          batch_size=256,
          epochs=10,
          shuffle=True)
action = model.predict(image)