5 Commits

Author SHA1 Message Date
liaoxingyu
fec7abc461 finish v0.2 ddp training 2020-07-06 16:57:43 +08:00
liaoxingyu
9bef814155 update visulize logger 2020-05-27 22:53:27 +08:00
liaoxingyu
84c733fa85 fix: remove prefetcher, put normalizer in model
1. remove messy data prefetcher which will cause  confusion
2. put normliazer in model to accelerate training via GPU computing
2020-05-25 23:39:11 +08:00
liaoxingyu
2ac55a7601 feat: update roc curve and TPR@FPR metric
support plot multiple ROC curves with different model
2020-05-20 14:29:33 +08:00
liaoxingyu
e344eae1cc feat: support plotting roc curve and compute auc score
ROC curve and AUC score will be help for thresholds
2020-05-19 20:45:26 +08:00