Use roc auc instead of accuracy

[ziw-liu]

Update metrics for knowledge distillation plots to be robust to class imbalance.

[edyoshikun]

@edyoshikun replace accuracy for ROC AUC for dynaclr