3. Visualization¶
In [7]:
# Visualization of the model predictions on our training set
plt.figure(figsize=(12,6))
plt.subplot(121)
plt.scatter(X_train[:,0], X_train[:,1], c=y_train ,cmap=cm.coolwarm)
plt.title('Training set')
plt.axis('equal');
# Visualization of the model predictions on our test set
plt.subplot(122)
plt.scatter(X_test[:,0], X_test[:,1], c=prediction_values[:,0], cmap=cm.coolwarm)
plt.title('Model predictions on our Test set')
plt.axis('equal')
Out[7]:
