学習したデータの予測データの出力について

Question

４値分類をしていて学習までは済ませたのですが初学者で予測したものをどう出力したらいいかわかりません。教えていただけないでしょうか？

losses = []
for epoch in range(20):
    all_loss = 0

    for idx, batch in enumerate(train_iter):
        batch_loss = 0
        encoder.zero_grad()
        classifier.zero_grad()

        text_tensor = batch.Text[0]
        label_tensor = batch.Label
        out = encoder(text_tensor)
        score, attn = classifier(out)
        batch_loss = loss_function(score, label_tensor)
        batch_loss.backward()
        optimizer.step()
        all_loss += batch_loss.item()
    print("epoch", epoch, "\t" , "loss", all_loss)

epoch 0 	 loss 0.00016928317199926823
epoch 1 	 loss 0.00017767107419786043
epoch 2 	 loss 0.00017765220218279865
epoch 3 	 loss 0.00015627022912667599
epoch 4 	 loss 0.00015667204934288748
epoch 5 	 loss 0.00016959857384790666
epoch 6 	 loss 0.0001593413580849301
epoch 7 	 loss 0.00017441319778299658
epoch 8 	 loss 0.00013102027696731966
epoch 9 	 loss 0.00014183222447172739
epoch 10 	 loss 0.00012683556633419357
epoch 11 	 loss 0.00011949731924687512
epoch 12 	 loss 0.00011920673750864808
epoch 13 	 loss 0.00010974999077006942
epoch 14 	 loss 0.00010862096041819314
epoch 15 	 loss 0.00011654233640001621
epoch 16 	 loss 0.00011608449949562782
epoch 17 	 loss 0.00010288159228366567
epoch 18 	 loss 0.00010046235911431722
epoch 19 	 loss 0.00010321225727238925

Answer 1

classifierの最終層の活性化関数はどうなってます？
softmaxになっている場合は、scoreの変数に4クラスの確率が出力されています。
なので確率が最も高いクラスが予測するクラスということです。

具体的には、scoreの行方向で最大値のindexを調べればOKです

以下のpredが予測ラベルになるかと思います。
ご確認おねがしいます。

_, pred = torch.max(score, 1)