【发布时间】:2021-09-01 11:16:12
【问题描述】:
当我试图保存我的模型时,它发生了。如果您能给我一些帮助,我将不胜感激,谢谢!我已经阅读了 StackOverflow 或 GitHub 问题中的一些问题,但找不到解决方案。
# add first layer with dropout
regressor.add(LSTM(units=50, return_sequences=True, input_shape=(X_train.shape[1],1)))
regressor.add(Dropout(0.2))
# add second layer
regressor.add(LSTM(units=50, return_sequences=True))
regressor.add(Dropout(0.2))
# add third layer
regressor.add(LSTM(units=50, return_sequences=True))
regressor.add(Dropout(0.2))
# add fourth layer
regressor.add(LSTM(units=50))
regressor.add(Dropout(0.2))
# the output layer
regressor.add(Dense(units=1))
# compiling the LSTM RNN network
regressor.compile(optimizer='rmsprop', loss='mean_squared_error')
# fit to the training set
regressor.fit(X_train, y_train, epochs=5, batch_size=32)
import pickle
filename = 'C:/Users/indore/Desktop/model.sav'
pickle.dump(regressor, open(filename, 'wb'))
【问题讨论】:
标签: python tensorflow keras lstm