scikitplot_roc_curve_with_sklearn_naive_bayes_classifier.py

1import matplotlib.pyplot as plt
2from sklearn.datasets import load_digits
3from sklearn.model_selection import train_test_split
4from sklearn.naive_bayes import GaussianNB
5import scikitplot as skplt
6
7# Load data
8X, y = load_digits(return_X_y=True)
9X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33)
10
11# Fit model
12nb = GaussianNB()
13nb.fit(X_train, y_train)
14
15# Predict
16predicted_probas = nb.predict_proba(X_test)
17
18# Plot
19skplt.metrics.plot_roc(y_test, predicted_probas)
20plt.show()