--- a +++ b/(2) PyTorch_HistoTNet/util/accuracy.py @@ -0,0 +1,4 @@ +def accuracy(confusion_matrix): + diagonal_sum = confusion_matrix.trace() + sum_of_all_elements = confusion_matrix.sum() + return diagonal_sum / sum_of_all_elements