TensorFlow with skflow: AttributeError：‘模块’对象没有属性'saver_pb2‘

Question

我尝试从skflow运行iris_save_restore.py示例，根本没有更改代码。

然而，我有一个错误，我不知道如何进行。似乎它是在从保存的文件中恢复模型时发生的。

new_classifier = skflow.TensorFlowEstimator.restore('/tmp/skflow_examples/iris_custom_model')
Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
score = metrics.accuracy_score(y_test, new_classifier.predict(X_test))
print('Accuracy: {0:f}'.format(score))  File "/Users/hamayu/.pyenv/versions/anaconda2-2.5.0/lib/python2.7/site-packages/skflow/estimators/base.py", line 463, in restore
    estimator._restore(path)
  File "/Users/hamayu/.pyenv/versions/anaconda2-2.5.0/lib/python2.7/site-packages/skflow/estimators/base.py", line 401, in _restore
    saver_def = tf.python.training.saver_pb2.SaverDef()
AttributeError: 'module' object has no attribute 'saver_pb2'

我使用以下版本：

• anaconda2-2.5.0
• tensorflow 0.7 CPU-只适用于Mac
• skflow 0.1.0

我怎么才能修好它？

Answer 1

这看起来像是skflow中的一个bug :当前版本的skflow与最新版本的TensorFlow (0.7.0和更高版本)不兼容。我提交了一个拉请求来修复它。(编辑：拉请求现在已经合并，因此升级到最新版本的skflow也将修复一些问题。)

同时，您可以通过编辑"skflow/estimators/base.py"的第401行(堆栈跟踪中的完整路径)来修复它，并替换：

saver_def = tf.python.training.saver_pb2.SaverDef()

...with：

saver_def = tf.train.SaverDef()

Answer 2

作者今天刚刚解决了这个问题：https://github.com/tensorflow/skflow/issues/144。您必须从github中删除并重新安装skflow (最新版本)。