以下代码似乎没有错误,但无论是否在“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
1 回答
慕容708150
TA贡献1831条经验 获得超4个赞
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()
试试这个..!
添加回答
举报
0/150
提交
取消