'change seaborn scatter plot legend from number to text
Here is how I plot the decision boundary of my SVC` classifier.
X, y = make_classification(n_samples=100, n_features=2, n_redundant=0,
n_clusters_per_class=1, weights=[0.9], flip_y=0, random_state=1)
# fit svm model
svc_model = SVC(kernel='linear', random_state=32)
svc_model.fit(X, y)
# Constructing hyperplane using a formula.
w = svc_model.coef_[0]
b = svc_model.intercept_[0]
y_points = np.linspace(X[:, 1].min(), X[:, 1].max())
x_points = -(w[1] * y_points + b) / w[0]
# Plotting our two-features-space
sns.scatterplot(x=X[:, 0], y=X[:, 1], hue=y, s=50)
# Plotting a red hyperplane
plt.plot(x_points, y_points, c='r')
output:
OK, but I want to change the legend from [0,1]
to ['cat', 'dog']
since that's what the labels represent.
EDIT
Using plt.legend()
produces a legend with decision boundary as label:
Solution 1:[1]
Replace last line as...
plt.plot(x_points, y_points, c='r')
plt.legend(["Cat", "Dog"])
If you need the same shape, you will need to create custom legend. Show below...
ax=plt.plot(x, y, c='r')
from matplotlib.lines import Line2D
custom = [Line2D([], [], marker='.', color='blue', linestyle='None'),
Line2D([], [], marker='.', color='orange', linestyle='None')]
plt.legend(custom, ["Cat", "Dog"])
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
Solution | Source |
---|---|
Solution 1 |