Upload mc_auroc.py

NaN cases are now set to 0.

mc_auroc.py

@@ -117,6 +117,10 @@ class AVG_MULTICLASS_AUROC(evaluate.Metric):
 
             roc_auc[i] = auc(fpr[i], tpr[i])
 
+            # if AUC is NaN, set it to 0
+            if numpy.isnan(roc_auc[i]):
+                roc_auc[i] = 0
+
         # Compute average AUC
         average_auc = numpy.mean(list(roc_auc.values()))
         average_auc_ci_low = numpy.mean(list(roc_auc_ci_low.values()))