딥러닝 김예림 강사님 반

마지막 부분 코드입니다!

def plot_decision_boundary(model, 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

xx, yy = np.meshgrid(np.arange(x_min, x_max, 0.01),

np.arange(y_min, y_max, 0.01))

Z = model(torch.tensor(np.c_[xx.ravel(), yy.ravel()], dtype=torch.float32))

Z = Z.argmax(dim=1).reshape(xx.shape)

plt.contourf(xx, yy, Z, alpha=0.8)

plt.scatter(X[:, 0], X[:, 1], c=y, s=20, edgecolor='k')

plt.show()

plot_decision_boundary(model, X, y)