pycognaize.common.utils.ConfusionMatrix

class ConfusionMatrix(TP=0, TN=0, FP=0, FN=0)[source]

Bases: object

Parameters:

  • TP (int)

  • TN (int)

  • FP (int)

  • FN (int)

Methods

compute_metrics

Given confusion matrix, computes f1 score, precision,

draw_cm

to_dict

Attributes

FN

FP

TN

TP

accuracy

f1

precision

recall

compute_metrics()[source]
Given confusion matrix, computes f1 score, precision,

recall and accuracy.

Return type:

dict

Returns:

results in dict