2016-07-01 1 views
7

Я пытаюсь выполнить этот учебник по методу тензорного потока, который использует функцию load_csv. TUTORIAL_LINKГде задана цель, указанная в функции load_csv функции tensorflow

Один из двух линий в вопросе:

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, но я не знаю, как это указано как таковое.

Это подразумевается как последняя колонка?

ответ

12

Из code:

def load_csv(filename, target_dtype, target_column=-1, has_header=True): 
    """Load dataset from CSV file.""" 

по умолчанию для target_column является -1. Итак, последний столбец, хорошо знать.