我有一个用python编写的算法,它是使用LSTM进行时序分析的。我的教授要求我展示在代码中创建的模型的细节。我怎么检查这里的“模型”?它是否在背景中对模型进行了可视化?
model = Sequential()
model.add(LSTM(50, input_shape=(trainX.shape[1], trainX.shape[2])))
model.add(Dropout(0.2))
model.add(Dense(1))
model.compile(loss='mae', optimizer='adam')
history = model.fit(trainX, trainY, epochs=50, batch_size=72, validation_data=(testX, testY), verbose=0, shuffle=False)发布于 2018-04-19 09:09:01
Keras中有一个名为plot_model的可视化工具。您可以使用它将模型保存为图像,在其中可以看到模型的结构,包括输入和输出维度。
from keras.utils import plot_model
plot_model(model, to_file='model.png')你可以在这里读到更多关于它的信息:Keras可视化
https://stackoverflow.com/questions/49916517
复制相似问题