python-2.7 - 由于 AttributeError，Tensorflow 示例全部失败：“模块”对象没有属性“load_csv”

Question

我正在运行张量流 0.11。这是我刚刚从 TensorFlow 教程中复制来加载 CSV 数据集的代码。

代码

import tensorflow as tf
import numpy as np


IRIS_TRAINING = "iris_training.csv"
IRIS_TEST = "iris_test.csv"

# Load datasets.
training_set = tf.contrib.learn.datasets.base.load_csv(filename=IRIS_TRAINING, target_dtype=np.int)
test_set = tf.contrib.learn.datasets.base.load_csv(filename=IRIS_TEST,  target_dtype=np.int)

错误

AttributeError: 'module' object has no attribute 'load_csv'

我的问题是我在 mac 中运行了相同的示例，它运行良好，但是当我在 ubuntu 14.04 LTS 中运行它时，它显示了这个错误。有人可以帮助解决这个问题。

Answer 1

在尝试 tensorflow 关于 Iris 数据集的教程时，我在 mac 上遇到了同样的问题。当我阅读有关“tf.contrib.learn.datasets.base”的张量流代码时，函数的名称实际上是“load_csv_with_header”，带有三个参数。所以试试这个：

    #Load datasets
    training_set = tf.contrib.learn.datasets.base.load_csv_with_header(filename=IRIS_TRAINING, target_dtype=np.int, features_dtype=np.float32)
    test_set = tf.contrib.learn.datasets.base.load_csv_with_header(filename=IRIS_TEST, target_dtype=np.int, features_dtype=np.float32)

希望这可以解决问题。