ml001.py

# =========================================================
#
# From: www.youtube.com/watch?v=cKxRcEZd3Mw
# =========================================================

from sklearn import tree

##features = [[140,'smooth'],[130,'smooth'],[150,'buumpy'],[170,'bumpy']]
##labels = ['apple','apple','orange','orange']

features = [[140,1],[130,1],[150,0],[170,0]]
labels = [0,0,1,1]

clf = tree.DecisionTreeClassifier()
clf = clf.fit(features,labels)

print(clf.predict([[160,0]]))