質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.35%
PyTorch

PyTorchは、オープンソースのPython向けの機械学習ライブラリ。Facebookの人工知能研究グループが開発を主導しています。強力なGPUサポートを備えたテンソル計算、テープベースの自動微分による柔軟なニューラルネットワークの記述が可能です。

CNN (Convolutional Neural Network)

CNN (Convolutional Neural Network)は、全結合層のみではなく畳み込み層とプーリング層で構成されるニューラルネットワークです。画像認識において優れた性能を持ち、畳み込みニューラルネットワークとも呼ばれています。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

Q&A

解決済

1回答

1585閲覧

CNN CrossEntropyLoss モデル出力結果について

ruizu

総合スコア35

PyTorch

PyTorchは、オープンソースのPython向けの機械学習ライブラリ。Facebookの人工知能研究グループが開発を主導しています。強力なGPUサポートを備えたテンソル計算、テープベースの自動微分による柔軟なニューラルネットワークの記述が可能です。

CNN (Convolutional Neural Network)

CNN (Convolutional Neural Network)は、全結合層のみではなく畳み込み層とプーリング層で構成されるニューラルネットワークです。画像認識において優れた性能を持ち、畳み込みニューラルネットワークとも呼ばれています。

Python

Pythonは、コードの読みやすさが特徴的なプログラミング言語の1つです。 強い型付け、動的型付けに対応しており、後方互換性がないバージョン2系とバージョン3系が使用されています。 商用製品の開発にも無料で使用でき、OSだけでなく仮想環境にも対応。Unicodeによる文字列操作をサポートしているため、日本語処理も標準で可能です。

0グッド

0クリップ

投稿2020/10/06 08:30

編集2020/10/06 10:40

前提・実現したいこと

機械学習を学び始めたばかりで、pytorchを使ってcnnモデルを作成しました。そこで損失関数にクロスエントロピー、最適化をadamに設定しました。
クロスエントロピーには標準でsoftmaxが入っているために、最後にそのままで0-1の確率値がでると学んだと思ったのですが、実際に確認するとマイナスが含まれています。

理解をし間違えていると思うのですが、なぜmodel()の結果にマイナスがあるのでしょうか
また、今回のモデルのように最終的に分類した20種それぞれである確率値を出力したい場合どのようにしたらいいのでしょうか

発生している問題

######モデル

python

1class cnn(nn.Module): 2 def __init__(self): 3 super().__init__() 4 self.conv1 = nn.Conv2d(3, 10, 3, 1) 5 self.conv2 = nn.Conv2d(10, 20, 3, 1) 6 self.fc1 = nn.Linear(20*6*6, 360) 7 self.fc2 = nn.Linear(360, 100) 8 self.fc3 = nn.Linear(100, 20) 9 10 self.criterion = nn.CrossEntropyLoss() 11 self.optimizer = torch.optim.Adam(self.parameters(), lr = 1e-3) 12 13 def forward(self, x): 14 x = F.relu(self.conv1(x)) 15 x = F.max_pool2d(x, 2, 2) 16 x = F.relu(self.conv2(x)) 17 x = F.max_pool2d(x, 2, 2) 18 x = x.view(-1, 20*6*6) 19 x = F.relu(self.fc1(x)) 20 x = F.relu(self.fc2(x)) 21 x = self.fc3(x) 22 return x

######出力

python

1 2   def train(model, data_loader): 3   model.train() 4       #省略 5 for X_train, y_train in data_loader: 6 7 outputs = model(X_train) 8       #省略 9 10      #出力結果例 11      #tensor([[ 0.0622, 0.0024, 0.1191, ..., -0.0031, -0.0571, 0.0554], 12

このoutputsを見てマイナスがでています。

試したこと

活性化関数やモデルの作り方について調べる

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

回答1

0

自己解決

model()の出力結果は生値であるため、出力結果をさらにsoftmaxに入れることで0-1の値に収めることが出来る

投稿2020/10/08 07:22

ruizu

総合スコア35

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.35%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問