2016-07-01 18 views
7

Sto provando a seguire questo tutorial di tensorflow che utilizza una funzione load_csv. TUTORIAL_LINKDove è specificato il target nella funzione load_csv di tensorflow

Una delle due linee in questione è:

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

Dove "iris_test.csv" assomiglia:

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 

Sono abbastanza sicuro che l'obiettivo del esercizio di apprendimento macchina è il verginica colonna ma non ho idea di come sia specificato come tale.

È implicito come ultima colonna?

risposta

12

Dal code:

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

predefinito per target_column è -1. Quindi, ultima colonna, buono a sapersi.