tensorflow中使用mnist数据集训练全连接神经网络 ——学习曹健老师“人工智能实践:tensorflow笔记”的学习笔记, 感谢曹老师 前期准备:mnist数据集下载,并存入data目录: 文件列表:四个文件,分别为训练和测试集数据 Four files are available on 官网 http://yann.lecun.com/exdb/mnist/ : train-images-idx3-ubyte.gz: training set images (9912422 by
Anaconda安装Keras: conda install keras 安装完成: 在Jupyter Notebook中新建并执行代码: import keras from keras.datasets import mnist # 从keras中导入mnist数据集 from keras.models import Sequential # 导入序贯模型 from keras.layers import Dense # 导入全连接层 from keras.optimizers import
import torch import numpy as np import torch.nn as nn from torch.autograd import Variable import torch.optim as optim from torch.utils.data import DataLoader from torchvision import datasets, transforms batch_size = 64 learning_rate = 1e-2 num_epoche