在python中分散数据并绘制决策边界

Question

我有这个代码，我已经加载了 iris 数据集，将数据分散到两个类中（以一个对所有的形式）我设计了一个单层感知器，W_ 是权重，绘图代码如下：

x = train_s[: , 1].T
y = train_s[: , 2].T
label = train_t
colors = ['green','blue']
plt.scatter(x, y, c=label, cmap=matplotlib.colors.ListedColormap(colors))
x1 = np.linspace(4,8)
y1 = np.linspace(1.5,5)
X,Y = np.meshgrid(x1,y1)
plt.plot([0,W_[1]],[0,W_[2]])
plt.show()

但是结果图是这样的：

如何解决这个问题？

Answer 1

你想用你的散点图跨越这条情节线：plt.plot([0,W_[1]],[0,W_[2]])？ 如果是这样，假设问题就在这个范围图中。 你的散点图有这个轴范围： X：4~8 Y：1.5~5 但是你的线图有范围 X:[0,W_[1]] Y:[0,W_[2]] 您应该“规范化”两个范围以获得良好的图形合并。