1. 程式人生 > >機器學習中,邏輯迴歸函式的簡單使用

機器學習中,邏輯迴歸函式的簡單使用

比如,現在要解決這樣一個問題。現在有5組資料,已知有一個20歲年收入3W的人不會買車,有一個23歲年收入7W的人會買車,有一個31歲年收入10W的人會買車,有一個50歲年收入7W的人不會買車,有一個60歲年收入5W的人不會買車,試求一個28歲年收入8W的人,買車的概率是多少?

我們用sklearn庫中的linear_model可以很容易的得到答案
程式碼實現過程如下:

from sklearn import linear_model
x = [[20,3],[23,7],[31,10],[42,13],[50,7],[60,5]]
y = [0,1,1,1,0,0]
lr = linear_model.LogisticRegression()
lr.fit(x,y)

testX = [[28,8]]
testY = lr.predict(testX)
prob = lr.predict_proba(testX)
print(prob)

列印結果如下:
在這裡插入圖片描述
其中0.1469是結果為0的概率(也就是不會買車的概率),會買車的概率為0.853