pytorch深度学习入门与实战孙玉林第六章卷积神经网络识别Fashion-MNIST.pdfVIP

pytorch深度学习⼊门与实战孙⽟林第六章卷积神经⽹络识别Fashion-MNIST

importnumpyasnp

importpandasaspd

fromsklearn.metricsimportaccuracy_score,confusion_matrix,classification_report

importmatplotlib.pyplotasplt

importseabornassns

importcopy

importtime

importtorch

importtorch.nnasnn

fromtorch.optimimportSGD,Adam

importtorch.utils.dataasData

fromtorchvisionimporttransforms

fromtorchvision.datasetsimportFashionMNIST

使⽤数据，准备训练数据集

#FashionMNIST

train_data=FashionMNIST(rootroot