# 只适合2D
def plot_decision_boundary(model, X, y):
# 求点中 最大值,最小值(点的形式为(x, y))
x_min, x_max = X[0, :].min() - 1, X[0, :].max() + 1
y_min, y_max = X[1, :].min() - 1, X[1, :].max() + 1
h = 0.01
# 生成网格点的操作
xx, yy = np.meshgrid(np.arange(x_min, x_max, h), np.arange(y_min, y_max, h))
# 用来预测模型 predict
Z = model(np.c_[xx.ravel(), yy.ravel()])
Z = Z.reshape(xx.shape)
# 填充等高线之间的内容
plt.contourf(xx, yy, Z, cmap=plt.cm.Spectral)
plt.ylabel('x2')
plt.xlabel('x1')
plt.scatter(X[0, :], X[1, :], c=y, cmap=plt.cm.Spectral)
画决策界限
猜你喜欢
转载自blog.csdn.net/clearlove_black/article/details/84915036
今日推荐
周排行