我正在处理一个非常简单的示例。我在同一组轴上创建了三个散点图,我绘制的每个数据集都有一个不同的关联颜色图。然而,传说并不像我想要的那样;这是为什么?
import matplotlib.pyplot as plt
from sklearn.datasets import make_blobs
X, y = make_blobs(n_samples=500, n_features=2, cluster_std=1.0, centers=[(0,0), (3,3)])
plt.scatter(X[:, 0], X[:, 1], c=y, s=50, cmap='rainbow');
clf_tree = DecisionTreeClassifier(criterion='entropy', max_depth=1)
clf_tree = clf_tree.fit(X, y)
y_pred = clf_tree.predict(X)
clf_tree = DecisionTreeClassifier(criterion='entropy', min_samples_leaf = 3)
clf_tree = clf_tree.fit(X, y)
y_pred = clf_tree.predict(X)
clf_tree = DecisionTreeClassifier(criterion='entropy', max_leaf_nodes = 3)
clf_tree = clf_tree.fit(X, y)
y_pred = clf_tree.predict(X)
#shap.decision_plot(expected_value, sh, features_display, link='logit', highlight=misclassified)
for i in range(len(y)):
if y[i] != y_pred[i]:
plt.scatter(X[i, 0], X[i, 1], c=y[i], s=50, cmap='Dark2')
plt.legend(("0","1","miss"))