tf.contrib.学习加载不在TensorFlow 1.1中工作的_csv_

Question

我安装了最新的TensorFlow（v1.1.0），并尝试运行tf.contrib.learn Quickstart教程，在这里您假设要为IRIS数据集构建一个分类器。但是，当我尝试：

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

我得到了一个StopIteration错误。在

当我检查API时，我没有找到关于load_csv_with_header()的任何信息。他们是否在没有更新教程的情况下更改了最新版本？我该怎么解决这个问题？在

编辑： 如果这有什么不同的话，我就用Python3.6。在

Answer 1

或者您可以将csv文件写为二进制文件，而不是添加decode()

if not os.path.exists(IRIS_TRAINING):
    raw = urllib.request.urlopen(IRIS_TRAINING_URL).read()
    with open(IRIS_TRAINING, 'wb') as f:
        f.write(raw)