代码按照“python神经网络编程”这本书写的。
训练集和测试集下载链接
#构建自己的神经网络
#建立一个三层的神将网络
import numpy
#导入sigmod函数库
import scipy.special
#导入画图库
import matplotlib.pyplot as plt
#确保图片在这个页面,不是额外的窗口
%matplotlib inline
#神经网络类定义
class neuralNetwork:
#初始化神经网络
def __init__(self,inputnodes,hiddennodes,outputnodes,learningrate):
#设置每一层输入输出隐藏层的节点个数
self.inodes=inputnodes
self.hnodes=hiddennodes
self.onodes=outputnodes
#学习率
self.lr=learningrate
#链接权重矩阵wih和who
self.wih=(numpy.random.rand(self.hnodes,self.inodes)-0.5)
self.who=(numpy.random.rand(self.onodes,self.hnodes)-0.5)
#激活函数采用sigmod函数
self.activation_function=lambda x: scipy.special.expit(x)
pass
#训练神经网络
def train(self,inputs_list,targets_list):
#将输入转换为二维数组