더북(TheBook)
>>> from sklearn.model_selection import cross_val_predict
>>> preds = cross_val_predict(estimator=pipe_lr,
...                           X=X_train, 
...                           y=y_train,
...                           cv=10, 
...                           n_jobs=-1)
>>> preds[:10]
array([0, 0, 0, 0, 0, 0, 0, 1, 1, 1])

method 매개변수에 반환될 값을 계산하기 위한 모델의 메서드를 지정할 수 있습니다. 예를 들어 method='predict_ proba'로 지정하면 예측 확률을 반환합니다. 'predict', 'predict_proba', 'predict_log_proba', 'decision_ function' 등이 가능하며 기본값은 'predict'입니다.

>>> from sklearn.model_selection import cross_val_predict
>>> preds = cross_val_predict(estimator=pipe_lr,
...                           X=X_train, 
...                           y=y_train,
...                           cv=10, 
...                           method='predict_proba', 
...                           n_jobs=-1)
>>> preds[:10]
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.