#!dt/bin/python3
 from sklearn import tree
 import pandas as pd

 df = pd.read_csv("trips-learn.csv",
         header = 0)

 Y = df['driver']
 X = df[['dow', 'miles', 'brakes', 'accels',
         'speed', 'vehicle']]

 clf = tree.DecisionTreeClassifier()
 clf = clf.fit(X, Y)

    # Check predictions
 for input in [[6,2000,0,2,20,1],
               [6,2000,0,2,0,1]]:
     print( input, ":",
            clf.predict( [input] ) )