Skip to content

Instantly share code, notes, and snippets.

@bertomartin
Forked from alejandrofloresm/hello-world.py
Created April 17, 2016 04:34
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bertomartin/834152963690b2fcd8c1465690247297 to your computer and use it in GitHub Desktop.
Save bertomartin/834152963690b2fcd8c1465690247297 to your computer and use it in GitHub Desktop.
Hello World for Google
# Code example for:
# Hello World - Machine Learning Recipes #1 - Google Developers
# https://www.youtube.com/watch?v=cKxRvEZd3Mw
from sklearn import tree
# Bumpy = 0, Smooth = 1
features = [[140, 1], [130, 1], [150, 0], [170, 0]]
# Apple = 0, Orange = 1
labels = [0, 0, 1, 1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(features, labels)
print clf.predict([[160, 0]])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment