From b9ea7d22c6a87331c6c12edf3b62f495019a6182 Mon Sep 17 00:00:00 2001
From: Florin Andrei <901867+FlorinAndrei@users.noreply.github.com>
Date: Thu, 7 Dec 2023 14:28:06 -0800
Subject: [PATCH] Update logistic_visualize.py
With the current version of Matplotlib, I get white dots for the 0 class, which are hard to see.
This change uses the standard Matplotlib color palette for the classes, shifted by 1.
---
logistic_regression_class/logistic_visualize.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/logistic_regression_class/logistic_visualize.py b/logistic_regression_class/logistic_visualize.py
index 65289535..ba84ffe7 100644
--- a/logistic_regression_class/logistic_visualize.py
+++ b/logistic_regression_class/logistic_visualize.py
@@ -44,7 +44,10 @@ def sigmoid(z):
z = Xb.dot(w)
Y = sigmoid(z)
-plt.scatter(X[:,0], X[:,1], c=T, s=100, alpha=0.5)
+# make colors more visible
+plt_colors = [f'C{i+1}' for i in T]
+
+plt.scatter(X[:,0], X[:,1], c=plt_colors, s=100, alpha=0.5)
x_axis = np.linspace(-6, 6, 100)
y_axis = -x_axis
pFad - Phonifier reborn
Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.
Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.
Alternative Proxies:
Alternative Proxy
pFad Proxy
pFad v3 Proxy
pFad v4 Proxy