在matplotlib中使用三个数据集的决策边界

时间:2014-10-18 16:14:01

标签: python matplotlib machine-learning

我正在尝试确定给定数据集的判别函数和绘图决策边界。我的数据集有三个观察结果,比如说X1,X2和X3有两个特征。我能够绘制如下所示的数据集

Plot of Training data

现在我能够计算数据的均值和协方差,从而得出P(x | C)函数。但是当我尝试使用以下方法绘制此函数时,我得到了错误的结果。

Decision

我使用以下方式绘制决策边界:

x_min, x_max = XX[:, 0].min() - 1, XX[:, 0].max() + 1
y_min, y_max = XX[:, 1].min() - 1, XX[:, 1].max() + 1
xx, yy = meshgrid(arange(x_min, x_max, .02),
                     arange(y_min, y_max, .02))
Z = A*(xx**2)-B*(yy**2)-C*xx*yy-D*xx-E*yy+F
plt.contourf(xx, yy, Z)

我很感激任何纠正错误的指导。

先谢谢。

沙善

0 个答案:

没有答案