以下代码看似没有错误,但无论是否在“model.summarize”语句前加上打印语句,都无法显示模型。我对此很新手,所以这段代码可能很糟糕。编辑:我对此很新手,我只想验证我有一个工作模型,并使用摘要工具。如果我遗漏了很多,你知道有什么好的文档吗?
from keras import layersfrom keras.models import Sequentialfrom sys import setrecursionlimitfrom keras.optimizers import Adamsetrecursionlimit(10000)class DQN: def __init__(self, env): self.epsilon = 0.1 self.hidden_layers = 4 self.shape = (None, 200) self.input_units = 256 self.ouput_units = 256 self.hidden_units = 512 self.learning_rate = 0.005 def get_model(self): model = Sequential() model.add(layers.GRU(self.input_units, input_shape=self.shape, return_sequences=True)) for layer in range(self.hidden_layers): model.add(layers.GRU(self.hidden_units, return_sequences=True)) model.add(layers.GRU(self.ouput_units, return_sequences=True)) model.summary
回答:
class DQN: def __init__(self, env): pass def get_model(self): model = Sequential() model.add(layers.GRU(self.input_units, input_shape=self.shape, return_sequences=True)) for layer in range(self.hidden_layers): model.add(layers.GRU(self.hidden_units, return_sequences=True)) model.add(layers.GRU(self.ouput_units, return_sequences=True)) print(model.summary())target = DQN()target.get_model()
试试这个吧..!