Merge pull request #116 from boyinggong/dev

updated indeces for ROC curve
berkeley-stat159 · Dec 12, 2015 · af5c9ba · af5c9ba
2 parents 122e406 + d915ff2
commit af5c9ba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/code/scripts/logistic.py b/code/scripts/logistic.py
@@ -49,7 +49,7 @@
     logreg_proba = logreg.predict_proba(X)
     confusion = create_confusion(logreg_proba, y)
     addsub = fig.add_subplot(4, 4, i)
-    addsub, AUC = plot_roc(confusion, addsub)
+    addsub, AUC = plot_roc(confusion, addsub, i)
     Min_thrs = np.append(Min_thrs, getMin_thrs(confusion))
     AUC_smr = np.append(AUC_smr, AUC)
 

diff --git a/code/utils/logistic_function.py b/code/utils/logistic_function.py
@@ -60,7 +60,7 @@ def getMin_thrs(confusion):
     return thrs, false_pos, false_neg
 
 
-def plot_roc(confusion, fig):
+def plot_roc(confusion, fig, sub_i):
     """
         function to plot the ROC (receiver operating characteristic) curve and
         calculate the corresponding AUC (Area Under Curve).
@@ -97,5 +97,5 @@ def plot_roc(confusion, fig):
         AUC += (ROC[i+1,0]-ROC[i,0]) * (ROC[i+1,1]+ROC[i,1])
     AUC *= -0.5
 
-    plt.title('ROC curve, AUC = %.4f'%AUC)
+    plt.title('subject '+ str(sub_i)+', AUC = %.4f'%AUC)
     return fig, AUC