建立第一個簡單的AI分類器

bzt820801發表於2017-08-21
AI
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]))

相關文章