1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | x_min,x_max = X[:, 0 ]. min () - . 5 ,X[:, 0 ]. max () + . 5
y_min,y_max = X[:, 1 ]. min () - . 5 ,X[:, 1 ]. max () + . 5 xx,yy = np.meshgrid(np.arange(x_min,x_max, 0.2 ),np.arange(y_min,y_max, 0.2 ))
Z = gnb.predict_proba(np.c_[xx.ravel(),yy.ravel()])[:, 1 ]
Z = Z.reshape(xx.shape)
plt.contourf(xx,yy,Z,cmap = plt.cm.summer,alpha = . 8 )
plt.scatter(X_train[:, 0 ],X_train[:, 1 ],c = y_train,cmap = plt.cm.cool,edgecolor = 'k' )
plt.scatter(X_test[:, 0 ],X_test[:, 1 ],c = y_test,cmap = plt.cm.cool,edgecolor = 'k' ,alpha = 0.6 )
plt.xlim(xx. min (),xx. max ())
plt.ylim(yy. min (),yy. max ())
plt.xticks(())
plt.yticks(())
plt.show()
|