欢迎光临散文网 会员登陆 & 注册

softmax回归的从零实现

2021-12-01 12:07 作者:如果我是泡橘子  | 我要投稿

1、导包

Downloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-images-idx3-ubyte.gzDownloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-images-idx3-ubyte.gz to ../data\FashionMNIST\raw\train-images-idx3-ubyte.gz

100.0%

Extracting ../data\FashionMNIST\raw\train-images-idx3-ubyte.gz to ../data\FashionMNIST\rawDownloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-labels-idx1-ubyte.gzDownloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/train-labels-idx1-ubyte.gz to ../data\FashionMNIST\raw\train-labels-idx1-ubyte.gz

100.6%

Extracting ../data\FashionMNIST\raw\train-labels-idx1-ubyte.gz to ../data\FashionMNIST\rawDownloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-images-idx3-ubyte.gzDownloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-images-idx3-ubyte.gz to ../data\FashionMNIST\raw\t10k-images-idx3-ubyte.gz

100.0%

Extracting ../data\FashionMNIST\raw\t10k-images-idx3-ubyte.gz to ../data\FashionMNIST\rawDownloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-labels-idx1-ubyte.gzDownloading http://fashion-mnist.s3-website.eu-central-1.amazonaws.com/t10k-labels-idx1-ubyte.gz to ../data\FashionMNIST\raw\t10k-labels-idx1-ubyte.gz

119.3%

Extracting ../data\FashionMNIST\raw\t10k-labels-idx1-ubyte.gz to ../data\FashionMNIST\raw


2、初始化模型参数


3、定义softmax操作

(tensor([[5., 7., 9.]]), tensor([[ 6.],         [15.]]))

(tensor([[0.4275, 0.1748, 0.0551, 0.1662, 0.1763],         [0.0295, 0.1304, 0.1590, 0.4909, 0.1902]]), tensor([1.0000, 1.0000]))


4、定义模型


5、定义损失函数

tensor([0.1000, 0.5000])

tensor([2.3026, 0.6931])


6、定义准确率

0.0149


7、训练

训练结果


8、预测

预测结果





----to be continued----

softmax回归的从零实现的评论 (共 条)

分享到微博请遵守国家法律