Task2 Softmax回归

Softmax回归

Softmax回归为了分类任务而设计，相较于线性回归多了softmax函数映射。为了估计所有可能类别的条件概率，我们需要一个有多个输出的模型，每个类别对应一个输出。在训练softmax回归模型后，给出任何样本特征，我们可以预测每个输出类别的概率。通常我们使用预测概率最高的类别作为输出类别。如果预测与实际类别（标签）一致，则预测是正确的。

实现过程

1. 导入数据

import torch
from torch import nn
from d2l import torch as d2l


batch_size = 256
train_iter, test_iter = d2l