以及用于加载保存模型的代码是:
def load_full_model(self, path_to_model):
self.model = load_model(path_to_model, custom_objects = {'Agent':Agent} )
注意训练期间使用了完全相同的代理类。
这很可能发生,因为您以一种不接受基础
__init__
类参数而覆盖Model
name
。因此,从配置中,该类期望像get_config
name
参数。应通过将必要的论点添加到
__init__
中,最简单的
**kwargs
:来解决这一点。
def __init__(self,number_of_outputs: int,number_of_hidden_units: int, **kwargs):
super(Agent,self).__init__(**kwargs)