Sklearn predict_proba - Predicting Class Probabilities

Python

To predict the class probability when using sklearn classification models we use the predict_proba function.

In the example below we have trained a logistic regression model and then use the predict_proba function to predict the class probabilities for each example in the test dataset.

 1|  from sklearn.linear_model import LogisticRegression
 2|  
 3|  #Initialise & Fit Model
 4|  model = LogisticRegression()
 5|  model.fit(X_train, y_train)
 6|  
 7|  # predict class probabilities for each example in X_test
 8|  y_pred_proba = model.predict_proba(X_test)
 9|  
10|
detro
Classification | Predict_proba
Did you find this snippet useful?

Sign up for free to to add this to your code library

Remove Stop Words from Text in DataFrame Column
Python

Pandas | Nltk | Nlp

149
LightGBM Custom Loss Function
Python

LightGBM | Loss function

132
Tuning XGBoost Hyperparameters with Grid Search
Python

XGBoost | Tuning

127
How to Convert DataFrame Values Into Percentages
Python

Lambda | Percentages | Pipe

119
How to Scale Data Using Standard Scaler But Keep Column Names
Python

Scaler | Standard

92
LightGBM Hyperparameter Tuning with GridSearch
Python

Python | LightGBM | Hyperparameter tuning | Gridsearch

91
Calculating Root Mean Squared Error (RMSE) with Sklearn and Python
Python

Rmse

86
Dynamically Create Columns in Pandas Dataframe
Python

Pandas | Columns

68
How to Train a Catboost Classifier with GridSearch Hyperparameter Tuning
Python

Catboost | Hyperparameter tuning

68
Overwriting a Python Logging File Instead of Appending
Python

Logging

65