我对机器学习框架和Python都是新手。我从https://github.com/xiaochus/TrafficFlowPrediction获取了一个keras-tensorflow项目的源代码,并且安装了所有正确版本的CUDA和Cudnn。但是,在加载GRU模型后出现了错误:
ValueError: GRU(reset_after=False)与GRU(reset_after=True)不兼容。
请问有人可以帮我吗?谢谢。看起来这个函数有重载选项。我应该添加一些选项比如reset_after来启用/禁用它吗?我只是在猜测。
lstm = load_model('TrafficFlowPrediction/model/lstm.h5')#在加载GRU模型时出错gru = load_model('TrafficFlowPrediction/model/gru.h5')
回答:
我通过更改创建模型的这一部分解决了这个问题
model.add(GRU(units[1], input_shape=(units[0], 1), return_sequences=True))
model.add(GRU(units[1], input_shape=(units[0], 1), return_sequences=True, reset_after=True))