[AI]/python.sklearn
머신러닝.sigmoid function(3D)
givemebro
2020. 4. 13. 19:18
반응형
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure(figsize=[10,8])
ax = Axes3D(fig)
a = np.arange(-4,12,0.2)
b = np.arange(-4,12,0.2)
xx, yy = np.meshgrid(a,b)
ax.plot_surface(xx, yy, model.coef_[0,0]*xx + model.coef_[0,1]*yy + model.intercept_[0],
shade=True, alpha=0.1, color='b')
ax.plot_wireframe(xx, yy, model.coef_[0,0]*xx + model.coef_[0,1]*yy + model.intercept_[0],
rstride=2, cstride=2, color='0.5')
ax.scatter(X[:,0], X[:,1], y, c=y, s=60)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('target')
#ax.view_init(60, 70)

from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure(figsize=[10,8])
ax = Axes3D(fig)
a = np.arange(-4,12,0.2)
b = np.arange(-4,12,0.2)
xx, yy = np.meshgrid(a,b)
ax.plot_surface(xx, yy, sigmoid(model.coef_[0,0]*xx + model.coef_[0,1]*yy + model.intercept_[0]),
shade=True, alpha=0.3, color='b')
ax.plot_wireframe(xx, yy, sigmoid(model.coef_[0,0]*xx + model.coef_[0,1]*yy + model.intercept_[0]),
rstride=2, cstride=2, color='0.5')
ax.scatter(X[:,0], X[:,1], y, c=y, s=60)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('target')
ax.view_init(20, -80)

반응형