机器学习——15 手写数字识别-小数据集
1.手写数字数据集
- from sklearn.datasets import load_digits
- digits = load_digits()
代码图:
结果图:
2.图片数据预处理
- x:归一化MinMaxScaler()
- y:独热编码OneHotEncoder()或to_categorical
- 训练集测试集划分
- 张量结构
代码图:
结果图:
3.设计卷积神经网络结构
- 绘制模型结构图,并说明设计依据。
代码图:
结果图:
4.模型训练
- model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
- train_history = model.fit(x=X_train,y=y_train,validation_split=0.2, batch_size=300,epochs=10,verbose=2)
代码图:
结果图:
补交作业:第一次作业因为电脑还没寄到因此没交。第六次因为电脑出故障拿去修了因此没交。