6. Visualization¶
Models predictions on our test dataset
In [22]:
plt.figure(figsize =(12,7))
plt.scatter(X_test[:,0], X_test[:,1], c=prediction_values_test[:, 0], cmap=cm.coolwarm)
plt.title('Model predictions on our Test Dataset')
plt.xlabel('feature 1')
plt.ylabel('feature 2')
plt.axis("equal")
plt.colorbar()
Out[22]:
