由于AttributeError，Tensorflow示例全部失败：'module'对象没有属性'load_csv'

Question

我正在运行Tensor flow 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'的tensorflow代码时，函数的名称实际上是'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)

希望这可以解决问题。