def test_pandas_confusion_cm_stats_animals(self):
y_true = ['rabbit', 'cat', 'rabbit', 'rabbit', 'cat', 'dog', 'dog', 'rabbit', 'rabbit', 'cat', 'dog', 'rabbit']
y_pred = ['cat', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'cat', 'rabbit', 'rabbit']
print("y_true: %s" % y_true)
print("y_pred: %s" % y_pred)
cm = ConfusionMatrix(y_true, y_pred)
assert isinstance(cm, pdml.confusion_matrix.LabeledConfusionMatrix)
assert isinstance(cm.stats(), OrderedDict)
assert cm.population == len(y_true) # 12
cm.print_stats()
cm_stats = cm.stats() # noqa
assert cm.binarize("cat").TP == cm.get("cat") # cm.get("cat", "cat")
assert cm.binarize("cat").TP == 3
assert cm.binarize("dog").TP == cm.get("dog") # 1
assert cm.binarize("rabbit").TP == cm.get("rabbit") # 3
def test_pandas_confusion_cm_stats_animals(self):
y_true = ['rabbit', 'cat', 'rabbit', 'rabbit', 'cat', 'dog', 'dog', 'rabbit', 'rabbit', 'cat', 'dog', 'rabbit']
y_pred = ['cat', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'cat', 'rabbit', 'rabbit']
print("y_true: %s" % y_true)
print("y_pred: %s" % y_pred)
cm = ConfusionMatrix(y_true, y_pred)
assert isinstance(cm, pdml.confusion_matrix.LabeledConfusionMatrix)
assert isinstance(cm.stats(), OrderedDict)
assert cm.population == len(y_true) # 12
cm.print_stats()
cm_stats = cm.stats() # noqa
assert cm.binarize("cat").TP == cm.get("cat") # cm.get("cat", "cat")
assert cm.binarize("cat").TP == 3
assert cm.binarize("dog").TP == cm.get("dog") # 1
assert cm.binarize("rabbit").TP == cm.get("rabbit") # 3
def test_pandas_confusion_cm_binarize(self):
y_true = ['rabbit', 'cat', 'rabbit', 'rabbit', 'cat', 'dog', 'dog', 'rabbit', 'rabbit', 'cat', 'dog', 'rabbit']
y_pred = ['cat', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'cat', 'rabbit', 'rabbit']
cm = ConfusionMatrix(y_true, y_pred)
assert isinstance(cm, pdml.confusion_matrix.LabeledConfusionMatrix)
print("Confusion matrix:\n%s" % cm)
select = ['cat', 'dog']
print("Binarize with %s" % select)
binary_cm = cm.binarize(select)
print("Binary confusion matrix:\n%s" % binary_cm)
assert cm.sum() == binary_cm.sum()
def test_pandas_confusion_cm_binarize(self):
y_true = ['rabbit', 'cat', 'rabbit', 'rabbit', 'cat', 'dog', 'dog', 'rabbit', 'rabbit', 'cat', 'dog', 'rabbit']
y_pred = ['cat', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'cat', 'rabbit', 'rabbit']
cm = ConfusionMatrix(y_true, y_pred)
assert isinstance(cm, pdml.confusion_matrix.LabeledConfusionMatrix)
print("Confusion matrix:\n%s" % cm)
select = ['cat', 'dog']
print("Binarize with %s" % select)
binary_cm = cm.binarize(select)
print("Binary confusion matrix:\n%s" % binary_cm)
assert cm.sum() == binary_cm.sum()
