/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

0 vote192 views1 answer
Pyqgis
Python
1 vote144 views1 answer
Variance of array python
Python
0 vote177 views1 answer
Performance of extend vs append loop
Python
31 votes176 views4 answers
Python remove nan rows
Python
1 vote185 views2 answers
Changing the number of ticks on a Matplotlib plot axis
Python
2 votes202 views5 answers
List append python 3
Python
0 vote165 views2 answers
Python selenium switch to window
Python
22 votes169 views6 answers
Readlines from file python
Python
0 vote195 views1 answer
Selenium press tab python
Python
0 vote259 views1 answer
Unzipping the value using zip() python
Python