使用sklearn的RandomForestClassifier做簡單的二分類

情景：基於相親網站男方提供的個人基本資料作爲輸入，以女方是否相親做標籤做訓練，用訓練出的模型預測女方是否相親

forest.py

#coding:utf-8

from sklearn.ensemble import RandomForestClassifier

#學歷 0:大專 2:碩士 1:本科
#"年齡", "身高", "年收入", '學歷'
X = [
        [25, 179, 15, 0],
        [33, 190, 19, 0],
        [28, 180, 18, 2],
        [25, 178, 18, 2],
        [46, 100, 100, 2],
        [40, 170, 170, 1],
        [34, 174, 20, 2],
        [36, 181, 55, 1],
        [35, 170, 25, 2],
        [30, 180, 35, 1],
        [28, 174, 30, 1],
        [29, 176, 36, 1],
        ]

#有否相親 0:N 1:Y
y = [0, 1, 1, 1, 0, 0, 1, 0, 1, 1, 0, 1]

#現在我們把訓練數據，和對應的分類放入分類器中進行訓練
clf = RandomForestClassifier().fit(X, y)

#預測下面此人是否相親
p = [[28, 180, 18, 2]]
print(clf.predict(p))

#輸出結果是[1] 是