2

iris_save_restore.pyコードをまったく変更せずに、skflow から例を実行しようとしました。

しかし、エラーが発生し、続行する方法がわかりません。保存したファイルからモデルを復元する際に発生したようです。

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 Mac OS X の CPU のみ
  • スクフロー 0.1.0

どうすれば修正できますか？

4

2 に答える 2

4

これは skflow のバグのようです。現在のバージョンの skflow は、最新バージョンの TensorFlow (0.7.0 以降) と互換性がありません。プルリクエストを送信して修正しました。(編集:プル リクエストは現在マージされているため、最新バージョンの skflow にアップグレードすると問題も修正されます。)

"skflow/estimators/base.py"それまでの間、 (スタックトレースのフル パス)の 401 行目を編集し、以下を置き換えることで修正できます。

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

...と：

saver_def = tf.train.SaverDef()
于 2016-03-04T07:02:57.713 に答える
0

著者は本日、この問題を修正しました: https://github.com/tensorflow/skflow/issues/144。github (最新リリース) から skflow を削除して再インストールする必要があります。

于 2016-03-11T17:20:01.000 に答える