文章標題

機器學習-1

來自b站的一個視頻:
機器學習1

題目大意

給出一些水果的數據(蘋果和橘子),數據中包含了重量和表皮光滑度
要求生成classifier(分類器)來預測輸入的數據是蘋果還是橘子

python代碼:

# coding=utf-8
# 引入sklearn包
from sklearn import tree

# 判斷是蘋果還說橘子
# 數據格式:[重量(單位:克),表皮光滑]
# 0表示光滑,1表示粗糙
features = [[140,0],[130,0],[150,1],[170,1]];
# 0表示蘋果,1表示橘子
labels = [0,0,1,1];

# 使用決策樹過濾器
clf =  tree.DecisionTreeClassifier()
# 通過數據訓練
clf = clf.fit(features,labels)
# 預測一個重量爲150克的,外表光滑的水果說蘋果還是橘子
print clf.predict([[150,0]])
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章