In [22]:
plt.figure(figsize =(10,7))
plt.scatter(X_train[:,0], X_train[:,1], c=prediction_values[:, 0], cmap=cm.coolwarm)
plt.title('Probability of our Training Dataset')
plt.xlabel('feature 1')
plt.ylabel('feature 2')
Out[22]:
Text(0,0.5,'feature 2')