(label_pred, label_true, num_classes)
| 58 | |
| 59 | |
| 60 | def _fast_hist(label_pred, label_true, num_classes): |
| 61 | mask = (label_true >= 0) & (label_true < num_classes) |
| 62 | hist = np.bincount( |
| 63 | num_classes * label_true[mask].astype(int) + |
| 64 | label_pred[mask], minlength=num_classes ** 2).reshape(num_classes, num_classes) |
| 65 | return hist |
| 66 | |
| 67 | |
| 68 | def evaluate(predictions, gts, num_classes): |