plt.subplot(1,2,1) plot_original_set(X_test, y_test, 'LightBlue', 'Orange', 'Test set') plt.subplot(1,2,2) plot_grid(parameters_final, X_test) plot_classified(X_test, Y_predictions_test, 'LightBlue', 'Orange', 'Classified', 'feature1' ) plt.show()