在张量流中创建训练和测试向量时出错

Question

我正在尝试重新创建代码以训练link-中给出的虹膜数据集 https://tf-docs.readthedocs.io/en/latest/tensorflow/g3doc/tutorials/tflearn/

但是，当我尝试运行代码的以下部分时，我遇到了valueerror：

# Data sets
IRIS_TRAINING = "iris_training.csv"
IRIS_TEST = "iris_test.csv"

# 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)

错误是： ValueError：int（）的无效文字，基数为10：“分隔符长度”

有人可以帮助解决此问题吗？

Answer 1

您可以使用熊猫

<float, int>

然后改用import pandas as pd df = pd.read_csv('iris_training.csv') dataset = df.values

from_tensor_slices

不涉及任何编码的简单解决方案就是使用文本编辑器手动删除标头，然后立即使用您的方法进行操作。