我一直在使用Tensorflow库进行教程练习。
我使用了这个示例。我将示例中的参数从n_classes = 15
改为n_classes = 2
,因为我只有两个类需要分类。
我读取数据的方式如下:
train = pandas.read_csv('tensorflow_feed/test/train_with_abs.csv', header=None)X_train, y_train = train[1], train[0]test = pandas.read_csv('tensorflow_feed/test/test_with_abs.csv', header=None)X_test, y_test = test[1], test[0]
但这引发了以下错误:
Total words: 35Traceback (most recent call last): File "/home/sumit/PycharmProjects/experiments/text_classification_save_restore.py", line 94, in <module> classifier.fit(X_train, y_train) File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/estimators/base.py", line 160, in fit monitors=monitors) File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/estimators/estimator.py", line 449, in _train_model train_op, loss_op = self._get_train_ops(features, targets) File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/estimators/estimator.py", line 673, in _get_train_ops _, loss, train_op = self._call_model_fn(features, targets, ModeKeys.TRAIN) File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/estimators/estimator.py", line 656, in _call_model_fn features, targets, mode=mode) File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/learn/python/learn/estimators/base.py", line 369, in _model_fn predictions, loss = model_fn(features, targets) File "/home/sumit/PycharmProjects/experiments/text_classification_save_restore.py", line 73, in rnn_model word_list = tf.unpack(word_vectors, axis=1)TypeError: unpack() got an unexpected keyword argument 'axis'Process finished with exit code 1
回答:
“axis”参数是在6月23日才添加到tf.unpack中的,而你查看的示例已被修改以使用它:
https://github.com/tensorflow/tensorflow/commit/eff93149a6dc8e6826898fd9f9c28c81e21c9836
所以我建议你可以:
-
使用那个提交之前的旧版本示例,例如:https://github.com/tensorflow/tensorflow/blob/892ca4ddc12852a7b4633fd08f163941356cb4e6/tensorflow/examples/skflow/text_classification_save_restore.py
-
从github的最新版本构建一个新的Tensorflow。
希望这对你有帮助!