torchgeo icon indicating copy to clipboard operation
torchgeo copied to clipboard

All trainers should compute confusion matrices on at least the test split

Open calebrob6 opened this issue 4 years ago • 0 comments

  • [ ] Add torchmetrics confusion matrix to all trainers
  • [ ] Save the test results to a file in the experiment directory (end of train.py)

calebrob6 avatar Oct 01 '21 18:10 calebrob6