/nate

Network Auto Tester and Evaluator

Primary LanguagePython

nate

Network Auto Tester and Evaluator (NATE)

This tool allows to set up a list of network training tasks that would allow to conduct multiple experiments with different parameters. Automatically calculates accuracy scores (F-score, AUROC) for trained networks.

  • Language: Python
  • Framework: Pytorch
  • Supported tasks: Classification, Segmentation, Image generation
  • Network architectures: All torchvision nets, shallow convolution networks

For more information check the documentation located in doc/main.pdf