开始使用免费开始使用

从回归到多类别分类

您已经看到的二元分类、多类别分类和回归模型基本相似,只是模型上做了少量调整。

先构建一个用于回归的模型,然后对其进行微调,使其可以执行多类别分类

本练习是课程的一部分

使用 PyTorch 的深度学习入门

查看课程

交互式实操练习

通过完成这段示例代码来试试这个练习。

import torch
import torch.nn as nn

input_tensor = torch.Tensor([[3, 4, 6, 7, 10, 12, 2, 3, 6, 8, 9]])

# Implement a neural network with exactly four linear layers
model = nn.____(
  ____
)

output = model(input_tensor)
print(output)
编辑并运行代码