/Python/Plot roc curve for neural network keras

Plot roc curve for neural network keras

from sklearn.metrics import roc_curve, auc

history = model.fit(x_train, y_train, validation_data=(
        x_test, y_test), epochs=num_of_epochs, batch_size=batch_size, verbose=1)

y_pred = model.predict(x_test).ravel()

nn_fpr_keras, nn_tpr_keras, nn_thresholds_keras = roc_curve(y_test, y_pred)
auc_keras = auc(nn_fpr_keras, nn_tpr_keras)
plt.plot(nn_fpr_keras, nn_tpr_keras, marker='.', label='Neural Network (auc = %0.3f)' % auc_keras)

Similar codes

1 vote349 views1 answer
Jupyter notebook reload module
Python
14 votes328 views2 answers
How to check if a dictionary is empty in python
Python
1 vote392 views1 answer
How to create staff account in django
Python Django
5 votes340 views2 answers
Python string replace by index
Python
1 vote410 views2 answers
Python docstring use
Python
0 vote322 views1 answer
Get the name of a current script in python
Python
0 vote390 views1 answer
Remove outliers numpy array
Python
1 vote335 views1 answer
How to get pygame key
Python
0 vote327 views1 answer
Stemming words python
Python
1 vote305 views1 answer
How to delete image using python
Python