Args: pred (np.ndarray): binary array. label (np.ndarray): binary array of the same size.
(self, pred, label)
| 121 | self.corr_neg = 0 # correct predict negative |
| 122 | |
| 123 | def feed(self, pred, label): |
| 124 | """ |
| 125 | Args: |
| 126 | pred (np.ndarray): binary array. |
| 127 | label (np.ndarray): binary array of the same size. |
| 128 | """ |
| 129 | assert pred.shape == label.shape, "{} != {}".format(pred.shape, label.shape) |
| 130 | self.nr_pos += (label == 1).sum() |
| 131 | self.nr_neg += (label == 0).sum() |
| 132 | self.nr_pred_pos += (pred == 1).sum() |
| 133 | self.nr_pred_neg += (pred == 0).sum() |
| 134 | self.corr_pos += ((pred == 1) & (pred == label)).sum() |
| 135 | self.corr_neg += ((pred == 0) & (pred == label)).sum() |
| 136 | |
| 137 | @property |
| 138 | def precision(self): |