我想为这个二元响应数据使用不同的标记样式,但是我无法在此处集成它。我希望可以为一个类使用三角形标记,另一个类使用星形标记。同时,最好使用自定义图例,如1代表多数类,0代表少数类。感谢您的建议。谢谢!
import numpy as np
import matplotlib.pyplot as plt
from sklearn import svm
from sklearn.datasets import make_blobs
from numpy.random import seed
seed(133)
X, y = make_blobs(n_samples=[1000, 10],centers=[[0.0, 0.0], [2.0, 2.0]],cluster_std= [1.5, 0.5],random_state=0, shuffle=False)
colormap = np.array(['tab:orange', 'tab:blue'])
plt.scatter(X[:, 0], X[:, 1],s=40, c=colormap[y], cmap=plt.cm.Paired, edgecolors='k')