Как напечатать вывод sklearn confusion_matrix из функции?

Question

confusion_matrix работает должным образом из командной строки на моем ноутбуке, но я не могу заставить его печатать свои выходные данные, когда они записаны внутри функции. Он применяется к одним и тем же массивам в обоих случаях.

Я что-то упустил?

Answer 1

Если вы вызываете функцию в качестве последней команды в ячейке записной книжки, выходные данные будут напечатаны автоматически, однако это не тот случай, когда она заключена в функцию.В этом случае вам придется вызывать функцию print () где-то внутри функции.

import numpy as np
from sklearn.metrics import confusion_matrix

a = np.array([1,0,1,1,0,1])
b = np.array([0,1,0,1,0,0])

def show_confusion(a, b):
    print(confusion_matrix(a, b))

show_confusion(a, b)