我正在尝试确定给定数据集的判别函数和绘图决策边界。我的数据集有三个观察结果,比如说X1,X2和X3有两个特征。我能够绘制如下所示的数据集
现在我能够计算数据的均值和协方差,从而得出P(x | C)函数。但是当我尝试使用以下方法绘制此函数时,我得到了错误的结果。
我使用以下方式绘制决策边界:
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)
我很感激任何纠正错误的指导。
先谢谢。
沙善