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

时间:2018-07-08 18:54:48

标签: python plot scatter

我有此代码,我加载了虹膜数据集,将数据分为两类(一种形式对所有形式),我设计了一个单层感知器,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()

但是最终的数字是这样的:

https://i.stack.imgur.com/dZ5xu.png

如何解决?

1 个答案:

答案 0 :(得分:0)

您是否希望通过散点图越过该图线:plt.plot([0,W_[1]],[0,W_[2]])? 如果是这样,则假设问题在该范围图中。 您的散点图具有以下轴范围:  X:4〜8  Y:1.5〜5 但是你的线图有范围  X:[0,W_ [1]]  Y:[0,W_ [2]] 您应该“标准化”这两个范围,以实现良好的图形合并。