python - 加载的 keras 模型没有预测方法

Question

在DQN算法中，agent训练成功，模型保存为.h5文件。代理类有一个方法调用load如下：

def load(self):
    new_model = load_model('dqn_model.h5')
    if self.epsilon == 0.0:
       self.update_network_parameters()
    return new_model

def update_network_parameters(self):
    self.model.set_weights(self.new_model.get_weights())

我尝试通过创建new_model如下来运行评估步骤：

new_model = agent.load()

但new_model没有任何方法，如predict.

Answer 1

确保您在调用的类中导入了模型agent.load。

from keras.models import Model

另外，就我个人而言，model.predictIntellisense 中并没有出现，因此请确保代码在运行时实际上不会编译。如果是这样，请告诉我们异常错误是什么。