Getting Started with Machine Learning using SciKit-learning

ml1.py

#!/usr/bin/env python
# My first Machine Learning Program.

from sklearn import tree

"""
Apple & oranges problem [Weight, Texture]
"""
features1 = [
    [140, 0],
    [130, 0],
    [150, 1],
    [170, 1],
]

labels1 = [0, 0, 1, 1]

clf1 = tree.DecisionTreeClassifier()
clf1 = clf1.fit(features1, labels1)
print clf1.predict([[160, 0]])


"""
Sports car & minivan problem [Horse Power, # of Seats]
"""
features2 = [
    [1000, 2],
    [1500, 1],
    [800, 7],
    [1000, 5],
]

labels2 = [0, 0, 1, 1]

clf2 = tree.DecisionTreeClassifier()
clf2 = clf2.fit(features2, labels2)
print clf2.predict([[1200, 11]])