(self)
| 269 | torch.save(self.all_labels, self.save_path + '_labels.pth') |
| 270 | |
| 271 | def evaluate(self): |
| 272 | |
| 273 | topk = [1, 5, 10] |
| 274 | topk = [k for k in topk if k < self.all_preds.shape[-1]] |
| 275 | acc = accuracy(torch.from_numpy(self.all_preds), torch.from_numpy(self.all_labels), topk=topk) |
| 276 | |
| 277 | for k, a in zip(topk, acc): |
| 278 | print(f'Top{k} Acc: {a.item()}') |
| 279 | |
| 280 | |
| 281 | def get_acc_auroc_curves(logdir): |