类型错误:无法腌制_thread。RLock 对象 LSTM



当我试图保存我的模型时,它发生了。如果你能给我一些帮助,我会很感激的,谢谢!我在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')) 

您不必使用pickle来保存您的模型。使用内置方法regressor.save('model.h5')

最新更新