從迴歸到多類別分類
你先前看到的二元分類、多類別分類與迴歸模型,本質上都很相似,只是對模型做了少量調整。
先建立一個用於迴歸的模型,接著再調整這個模型,讓它能執行多類別分類。
本練習屬於課程
使用 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)