from sklearn import tree
# 第一個簡單的分類器
features = [[140, 1], [130, 1], [150, 0], [170, 0]] #列表左邊的變量代表水果的重量，右邊的1代表表面光滑,0代表有疙瘩
labels = [0, 0, 1, 1]                               #0代表蘋果，1代表橘子

clf = tree.DecisionTreeClassifier()   #創建一個分類器
clf = clf.fit(features, labels)       #給分類器定義一個規則，這裏使用到fit函數代表讓fit從數據中自己找到規律

print (clf.predict([155, 1]))

相關推薦