Skip to content

Commit

Permalink
add comment
Browse files Browse the repository at this point in the history
  • Loading branch information
boyinggong committed Dec 10, 2015
1 parent 863fff5 commit 4c69494
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions code/utils/logistic_function.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,15 @@
import matplotlib.pyplot as plt

def plot_roc(logreg_proba, y):
"""
function to plot the ROC (receiver operating characteristic) curve and
calculate the corresponding AUC (Area Under Curve).
Input:
logreg_proba: The estimate probability for each class calculated from
logistic regression model.
y: The actual class of response,.
Output: The ROC curve and and correspong AUC value.
"""

thresholds = np.linspace(1,0,101)

Expand Down

0 comments on commit 4c69494

Please sign in to comment.