def test_confusion_matrix():
"""Test ConfusionMatrix class"""
yactual = [2, 0, 2, 2, 0, 1, 1, 2, 2, 0, 1, 2]
ypred = [0, 0, 2, 1, 0, 2, 1, 0, 2, 0, 2, 2]
cm = ConfusionMatrix(yactual, ypred, "Sample_classifier")
number_cm = cm.get_number_cm()
normalized_cm = cm.get_normalized_cm()
metrics = {
"TP": cm.get_true_pos(),
"TN": cm.get_true_neg(),
"FP": cm.get_false_pos(),
"FN": cm.get_false_neg()
}
print("Confusion matrix:")
print(number_cm)
print("Normalized confusion matrix:")
print(normalized_cm)
print("Precision per label is:", cm.get_precision())
print("Metrics:", metrics)
print("MCC is", cm.get_mcc())
return cm.get_precision()
