/Keras-MLP

基于Keras搭建一个简单的多层感知机(MLP),实现手写数字识别

Primary LanguagePython

Keras-MLP

基于Keras搭建一个简单的多层感知机(MLP),用mnist数据集对MLP进行训练,完成模型的保存和加载和识别测试。

环境:
CUDA:11.6.134
cuDNN:8.4.0
keras:2.9.0
tensorflow:2.9.1

注意:
项目内目录中两个文件夹:
1./dataset:保存数据集
2./save_model:保存训练好的模型

Keras有三种构建model的方式:
1.Sequential序列模型
2.函数式API构建模型
3.子类subclass构建模型
不同方式在模型保存和加载方面方法不通用,按照自己习惯选择即可