2019-08-10 – Python量化投资

2019-08-10

CNN卷积神经网络

import torch

import torch.nn as nn   #为了简化,直接利用nn.nn来进行构建神经层

from torch.autograd import Variable

import torch.utils.data as Data

import torchvision   #一个库,里面包含了很多图片和数据的

import matplotlib.pyplot  as plt

#Hyper Parameters

EPOCH=1  #train the training data n times,to save time,we just train 1 epoch

……

train_data=torchvision.datasets.MNIST(    #MNIST是一个下载的代码,你可以去网上下载

root=’./mnist’,     #将这个保存,保存到root里面

https://www.jianshu.com/p/5f83fd837451

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论