我正在尝试使用这个使用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
列,但我不知道它是如何指定的那样。
它隐含在最后一栏吗?
答案 0 :(得分:12)
来自code:
def load_csv(filename, target_dtype, target_column=-1, has_header=True):
"""Load dataset from CSV file."""
<{1}}的默认值为target_column
。所以,最后一栏,很高兴知道。