1. 程式人生 > >機器學習:用6行Python程式碼開始寫第一個機器學習程式

機器學習:用6行Python程式碼開始寫第一個機器學習程式

import sklearn
from sklearn import tree
# features = [[140, "smooth"],[130, "smooth"],[150, "bumpy"],[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([[150, 0]]))