1. 程式人生 > >機器學習(一) 效果圖實現程式碼

機器學習(一) 效果圖實現程式碼

xmin, xmax = data[:,0].min(), data[:,0].max()
ymin, ymax = data[:,1].min(), data[:,1].max()

x = np.linspace(xmin,xmax, 1000)
y = np.linspace(ymin, ymax, 1000)
X,Y = np.meshgrid(x,y)

X_test = np.c_[X.ravel(), Y.ravel()]

y_logistic = logistic.predict(X_test)

plt.pcolormesh(X,Y,y_logistic.reshape(1000,1000))
plt.scatter(data[:, 0], data[:,
1], c=target, cmap='rainbow')

演算法模型實現效果圖