1. 程式人生 > >機器學習庫一:scikit-learn

機器學習庫一:scikit-learn

資料探勘中的機器學習庫scikit-learn,簡稱:sklearn

功能:分類、迴歸、降維、聚類四個機器學習演算法和特徵提取、資料處理、模型評估三個模組

安裝:pip install sklearn

決策樹sklearn.tree使用demo

from sklearn import tree
# 訓練資料
feature = [[178, 1], [155, 0], [177, 0], [165, 0], [169, 1], [169, 0]]
# 性別分類
label = ['男', '女', '男', '女', '男', '女']
# 建立決策樹分類器例項
clf = tree.DecisionTreeClassifier()
# 擬合訓練資料,得到訓練模型引數
clf = clf.fit(feature, label)
# 預測資料1,並輸出預測結果
s1 = clf.predict([[158, 0]])
print(u'第1個測試的人預計性別是:', s1[0])
# 預測資料2,並輸出預測結果
s2 = clf.predict([[176, 1]])
print(u'第2個測試的人預計性別是:', s2[0])

 結果: