import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
from sklearn.neural_network import MLPClassifier #神经网络分类
from sklearn.datasets import make_moons
from sklearn.model_selection import train_test_split
#产生样本
x,y=make_moons(100,noise=0.2)
x_train,x_test,y_train,y_test = train_test_split(x,y,train_size=0.6)
#构建神经网络
MLPC_model = MLPClassifier(hidden_layer_sizes=(6,3),max_iter=10000)
MLPC_model.fit(x_train,y_train)
#显示结果
x_min,x_max=x_train[:,0].min(),x_train[:,1].max()
y_min,y_max=x_train[:,1].min(),x_train[:,1].max()
#将x轴上的100个整数点组成的行向量,重复256次,构成shape(256,100)的X矩阵;将y轴上的256个整数点组成列向量,重复100次构成shape(256,100)的Y矩阵
xx,yy=np.meshgrid(np.linspace(x_min,x_max,200),np.linspace(y_min,y_max,200))#生成网格采样点 meshgrid(length(x),length(y))返回X行Y列的矩阵
grid_test = np.stack((xx.flat,yy.flat),axis=1) #测试点
y_predict = MLPC_model.predict(grid_test)
cm_pt = mpl.colors.ListedColormap(['w', 'k']) #样本点颜色
cm_bg = mpl.colors.ListedColormap(['r', 'y']) #背景颜色
plt.xlim(x_min, x_max);plt.ylim(y_min, y_max) #设置坐标范围
plt.pcolormesh(xx, yy, y_predict.reshape(xx.shape), cmap=cm_bg) #绘制网格背景
plt.scatter(x_train[:,0],x_train[:,1],c=y_train,cmap=cm_pt,marker='o') #绘制样本点
plt.grid(True);plt.show()
神经网络[分类]——机器学习
最新推荐文章于 2024-07-10 09:15:55 发布