Analyse de sentiment en Python
Violeta Misheva
Data Scientist
from sklearn.linear_model import LogisticRegression
# Regularization arguments
LogisticRegression(penalty='l2', C=1.0)
log_reg = LogisticRegression().fit(X_train, y_train)
# Prédire les étiquettes
y_predicted = log_reg.predict(X_test)
# Prédire la probabilité
y_probab = log_reg.predict_proba(X_test)
y_probab
array([[0.5002245, 0.4997755],
[0.4900345, 0.5099655],
...,
[0.7040499, 0.2959501]])
# Sélectionner les probabilités de la classe 1
y_probab = log_reg.predict_proba(X_test)[:, 1]
array([0.4997755, 0.5099655 ..., 0.2959501]])
# Encodage de probabilité par défaut :
# Si probabilité >= 0.5, alors classe 1 sinon classe 0
Analyse de sentiment en Python