人工智能与自动化 Python课程设计卷积神经网络手写数字识别系统源代码。
流程
参数配置文件的加载
with open(params_path) as params_file:
self.params = yaml.load((params_file))
这里使用了yaml文件作为配置文件,原因是yaml文件结构比较简单简洁,可以清楚地表示出层次结构,通过参数文件的配置,可以不用修改源代码就可以配置出不同的神经网络,具体见下面layer生成的分析
加载数据集,分割训练集和测试集
使用sklearn 来获取数据集,并且进行分割
# 加载数据
digits_data, digits_target = load_digits(
return_X_y=True)
# 划分训练集和测试集
self.x_train, self.x_test, self.y_train, self.y_test = \
train_test_split(digits_data, digits_target,
test_size=1-tr