數據挖掘中的機器學習庫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])

?結果：

?

總結

以上是生活随笔為你收集整理的机器学习库一：scikit-learn的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。