開始使用免費開始

從迴歸到多類別分類

你先前看到的二元分類、多類別分類與迴歸模型,本質上都很相似,只是對模型做了少量調整。

先建立一個用於迴歸的模型,接著再調整這個模型,讓它能執行多類別分類

本練習屬於課程

使用 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)
編輯並執行程式碼