Keras模型pyspark错误

时间:2017-09-29 04:43:00

标签: pyspark keras pickle

我有一个keras模型已被腌制,如下面的博客所述。

http://zachmoshe.com/2017/04/03/pickling-keras-models.html

奇怪的是,当我从一个html文件中运行模型时,从python中读取为open(filename),它按预期工作。但是当从pyspark读取时在文件上运行它时,我收到以下错误:

AttributeError("'Model' object has no attribute '_feed_input_names'",)

1 个答案:

答案 0 :(得分:0)

您还必须对每个工作人员运行make_keras_picklable()。否则,工作节点上__setstate__对象的Model方法不会更新,因此不会按预期反序列化。