Roc curve python

import sklearn.metrics as metrics
# calculate the fpr and tpr for all thresholds of the classification
probs = model.predict_proba(X_test)
preds = probs[:,1]
fpr, tpr, threshold = metrics.roc_curve(y_test, preds)
roc_auc = metrics.auc(fpr, tpr)

# method I: plt
import matplotlib.pyplot as plt
plt.title('Receiver Operating Characteristic')
plt.plot(fpr, tpr, 'b', label = 'AUC = %0.2f' % roc_auc)
plt.legend(loc = 'lower right')
plt.plot([0, 1], [0, 1],'r--')
plt.xlim([0, 1])
plt.ylim([0, 1])
plt.ylabel('True Positive Rate')
plt.xlabel('False Positive Rate')
plt.show()

# method II: ggplot
from ggplot import *
df = pd.DataFrame(dict(fpr = fpr, tpr = tpr))
ggplot(df, aes(x = 'fpr', y = 'tpr')) + geom_line() + geom_abline(linetype = 'dashed')

Source: stackoverflow.com

Similar codes

0 vote262 views1 answer
Read specific rows from csv in python
Python
39 votes246 views6 answers
Python check if int
Python
41 votes274 views4 answers
Find out current datetime in python
Python
44 votes270 views2 answers
Cm to foot
Python
2 votes247 views1 answer
How to find geometric mean in python
Python
1 vote260 views1 answer
Loss funfction suited for softmax
Python
0 vote253 views1 answer
Set permissions role discord.py
Python
3 votes270 views2 answers
Tables in python
Python
0 vote298 views1 answer
Dynamic footer in django
Python Django
0 vote254 views1 answer
Python create a dictionary of integers
Python