Pythonで学ぶSentiment Analysis
Violeta Misheva
Data Scientist
from sklearn.linear_model import LogisticRegression
# 正則化の引数
LogisticRegression(penalty='l2', C=1.0)
log_reg = LogisticRegression().fit(X_train, y_train)
# クラス予測
y_predicted = log_reg.predict(X_test)
# 確率予測
y_probab = log_reg.predict_proba(X_test)
y_probab
array([[0.5002245, 0.4997755],
[0.4900345, 0.5099655],
...,
[0.7040499, 0.2959501]])
# クラス1の確率のみ取得
y_probab = log_reg.predict_proba(X_test)[:, 1]
array([0.4997755, 0.5099655 ..., 0.2959501]])
# 既定の確率のクラス化:
# 確率 >= 0.5 ならクラス1、そうでなければクラス0
Pythonで学ぶSentiment Analysis