在tensorflow的load_csv函数中指定了target的位置

时间:2016-07-01 17:50:30

标签: python tensorflow

我正在尝试使用这个使用load_csv函数的tensorflow教程。 TUTORIAL_LINK

有问题的两条线之一是:

IRIS_TEST = "iris_test.csv"
test_set = tf.contrib.learn.datasets.base.load_csv(
                                                   filename=IRIS_TEST,
                                                   target_dtype=np.int
                                                  )

"iris_test.csv"的样子:

30,4,setosa,versicolor,virginica
5.9,3.0,4.2,1.5,1
6.9,3.1,5.4,2.1,2
5.1,3.3,1.7,0.5,0
6.0,3.4,4.5,1.6,1
5.5,2.5,4.0,1.3,1
6.2,2.9,4.3,1.3,1
5.5,4.2,1.4,0.2,0
6.3,2.8,5.1,1.5,2
5.6,3.0,4.1,1.3,1
6.7,2.5,5.8,1.8,2
7.1,3.0,5.9,2.1,2
4.3,3.0,1.1,0.1,0

我非常确定机器学习练习的目标是verginica列,但我不知道它是如何指定的那样。

它隐含在最后一栏吗?

1 个答案:

答案 0 :(得分:12)

来自code

def load_csv(filename, target_dtype, target_column=-1, has_header=True):
  """Load dataset from CSV file."""
<{1}}的默认值为target_column。所以,最后一栏,很高兴知道。