以下代码似乎没有错误,但无论是否在“model.summarize”语句之前有打印语句,都不会显示模型。我对此很陌生,所以这段代码可能非常糟糕。编辑:我再次对此感到陌生,我想做的就是验证我是否有一个带有摘要实用程序的工作模型。我是否错过了很多,如果是的话,你知道这方面有什么好的文档吗?
from keras import layers
from keras.models import Sequential
from sys import setrecursionlimit
from keras.optimizers import Adam
setrecursionlimit(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
慕容708150
相关分类