반응형
Notice
Recent Posts
Recent Comments
Link
관리 메뉴

bro's coding

sklearn.LogisticRegression.predict_proba 본문

[AI]/python.sklearn

sklearn.LogisticRegression.predict_proba

givemebro 2020. 4. 13. 11:09
반응형

https://broscoding.tistory.com/129

 

머신러닝.linear_model.LogisticRegression(로지스틱 회귀)

import numpy as np import matplotlib.pyplot as plt # data 준비 from sklearn.datasets import make_blobs X,y=make_blobs(400,2,[[0,0],[5,5]],[2,3]) https://broscoding.tistory.com/128 머신러닝.datasets..

broscoding.tistory.com

# predict_proba()
# proba = probablity(확률)
# 0일 확률 1일 확률 표시
np.round(model.predict_proba(X)[:10],2)
array([[0.94, 0.06],
       [0.89, 0.11],
       [0.14, 0.86],
       [0.06, 0.94],
       [0.  , 1.  ],
       [0.01, 0.99],
       [0.83, 0.17],
       [0.87, 0.13],
       [0.  , 1.  ],
       [0.  , 1.  ]])

 

pred_y[:10],y[:10]
(array([0, 0, 1, 1, 1, 1, 0, 0, 1, 1]),
array([0, 0, 1, 1, 1, 1, 1, 0, 1, 1]))
반응형
Comments