Classification report sklearn

Hi All, I want to discuss Classification report from sklearn in this post.

It is important to keep different metrics in mind when we are both training and evaluating models. Depending on the context, certain metrics will make more sense than others. The best thing is that the Classification report summarize it very well in a single tabular structure.

Scikit-learn does provide a convenience report when working on classification problems to give us a quick idea of the accuracy of a model using a number of measures.

The classification_report() function displays the precision, recall, f1-score and support for each class.

The classification_report function builds a text report showing the main classification metrics.

Lets see the same using an example:

Preliminaries

# Load libraries
from sklearn import datasets
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.metrics import classification_report

Load Iris Flower Data

# Load data
iris = datasets.load_iris()

# Create feature matrix
X = iris.data

# Create target vector
y = iris.target

# Create list of target class names
class_names = iris.target_names

# Create training and test set
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=1)

# Create logistic regression
classifier = LogisticRegression()

# Train model and make predictions
y_hat = classifier.fit(X_train, y_train).predict(X_test)

# Create a classification report
print(classification_report(y_test, y_hat, target_names=class_names))

             precision    recall  f1-score   support

     setosa       1.00      1.00      1.00        13
 versicolor       1.00      0.62      0.77        16
  virginica       0.60      1.00      0.75         9

avg / total       0.91      0.84      0.84        38

Happy Learning