混淆矩阵要求
要为整个数据集创建一个混淆矩阵,我们需要一个与训练集长度相同的一维预测张量。...更具体地说,混淆矩阵将显示模型正确预测的类别和模型不正确预测的类别。对于不正确的预测,我们将能够看到模型预测的类别,这将告诉我们哪些类别使模型混乱。...使用预测张量
现在,有了预测张量,我们可以将其传递给我们在上一节中创建的get_num_correct()函数以及训练集标签,以获取正确预测的总数。...这将创建一个充当热图的矩阵,告诉我们预测值相对于真实值的下降位置。
为此,我们需要具有目标张量和train_preds张量中的预测标签。...[ 0, 0, 0, 0, 1, 61, 0, 107, 1, 5830]],
dtype=int64)
PyTorch张量是类似于数组的