From 28ab7b04852e64e2e51cc50d948e1c89229d19c8 Mon Sep 17 00:00:00 2001
From: Soumyadip Sikdar <52605080+Soumyadip54321@users.noreply.github.com>
Date: Sat, 1 Jun 2024 14:39:40 +0530
Subject: [PATCH] Update helper_functions.py

---
 helper_functions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/helper_functions.py b/helper_functions.py
index 88f65521..e1657ebc 100644
--- a/helper_functions.py
+++ b/helper_functions.py
@@ -66,7 +66,7 @@ def plot_decision_boundary(model: torch.nn.Module, X: torch.Tensor, y: torch.Ten
     # Reshape preds and plot
     y_pred = y_pred.reshape(xx.shape).detach().numpy()
     plt.contourf(xx, yy, y_pred, cmap=plt.cm.RdYlBu, alpha=0.7)
-    plt.scatter(X[:, 0], X[:, 1], c=y, s=40, cmap=plt.cm.RdYlBu)
+    plt.scatter(X[:, 0], X[:, 1], c=y, s=40, cmap=plt.cm.gist_heat)
     plt.xlim(xx.min(), xx.max())
     plt.ylim(yy.min(), yy.max())