从回归到多类别分类
您已经看到的二元分类、多类别分类和回归模型基本相似,只是模型上做了少量调整。
先构建一个用于回归的模型,然后对其进行微调,使其可以执行多类别分类。
本练习是课程的一部分
使用 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)