NameError: name 'plot_confusion_matrix' is not defined

Question

I am trying to make a classification model using VGG16 but at the end of the project I faced an error for getting the Confusion Matrix. Below the codes are given,

Imported packages and modules are:

import os
import keras
import numpy as np
import tensorflow as tf
from keras.models import Model
import matplotlib.pyplot as plt
from keras.optimizers import Adam
from keras.applications import MobileNet
from sklearn.metrics import confusion_matrix
from keras.layers.core import Dense, Activation
from keras.metrics import categorical_crossentropy
from sklearn.model_selection import train_test_split
from keras.preprocessing.image import ImageDataGenerator
from keras.applications.mobilenet import preprocess_input
from tensorflow.keras.preprocessing import image_dataset_from_directory

N.B: For shorting I just skipping of dataset linked

Below define VGG16:

vgg16_model = keras.applications.vgg16.VGG16()
vgg16_model.summary()

Now, Define the model:

model = Sequential()
for layer in vgg16_model.layers:
  model.add(layer)

for layer in model.layers:
  layer.trainable = False

model.add(Dense(2, activation='softmax'))

Compile the model:

model.compile(Adam(lr=.0001), loss='categorical_crossentropy', metrics=['accuracy'])

Fit the model:

model.fit_generator(train_batches, steps_per_epoch=4, validation_data=valid_batches, validation_steps=4, epochs=10, verbose=2)

Now for confusion matrix:

test_imgs, test_labels = next(test_batches)
plots(test_imgs, titles=test_labels)
test_labels = test_labels[:,0] 
predictions = model.predict_generator(test_batches, steps=1, verbose=0)
cm = confusion_matrix(test_labels, np.round(predictions[:,0]))

Below I faced an error, Please concern below code,

cm_plot_labels = ['diseaseAffectedEggplant','freshEggplant']
plot_confusion_matrix(cm, cm_plot_labels, title="Confusion Matrix")   // this line, I faced an error

The Error is given below,

---------------------------------------------------------------------------
NameError                                 Traceback (most recent call last)
<ipython-input-28-43b96d543746> in <module>()
      1 cm_plot_labels = ['diseaseAffectedEggplant','freshEggplant']
----> 2 plot_confusion_matrix(cm, cm_plot_labels, title="Confusion Matrix")

NameError: name 'plot_confusion_matrix' is not defined

score 4 · Accepted Answer · answered Jan 10 '21 at 09:13

4

You need to import plot_confusion_matrix from the sklearn.metrics module:

from sklearn.metrics import plot_confusion_matrix

See documentation.

answered Jan 10 '21 at 09:13

Jarvis

8,494
3
27
58

thanks a lot but another error, I have faced after importing plot_confusion_matrix. That's is "TypeError: plot_confusion_matrix() got an unexpected keyword argument 'title'" – Imdadul Haque Jan 10 '21 at 09:26
1

@ImdadulHaque Remove title parameter it doesnt support that – Sachin Rajput Jan 10 '21 at 09:33

NameError: name 'plot_confusion_matrix' is not defined

1 Answers1