2016-07-01 15 views
7

load_csv işlevini kullanan bu tensnorlow öğreticisini takip etmeye çalışıyorum. Söz konusu iki hat arasında TUTORIAL_LINKHedef, tensorflow'un load_csv işlevinde nerede belirtiliyor

biri şudur: "iris_test.csv" benziyor

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

:

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 

Ben makine öğrenme egzersiz hedefidir eminim verginica sütun, ancak nasıl belirtildiğini bilmiyorum.

Son sütun olarak mı ima ediliyor? code itibaren

cevap

12

: target_column için

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

varsayılan -1 olduğunu. Yani, son sütun, bilmek güzel.