回帰から多クラス分類へ
これまで見てきたバイナリ分類、多クラス分類、回帰のモデルは、モデルに少し手を入れるだけでどれもよく似ていました。
まずは回帰用のモデルを作成し、その後で多クラス分類ができるようにモデルを調整しましょう。
この演習はコースの一部です
PyTorchで学ぶIntroduction to Deep Learning
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
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)