前提・実現したいこと
pytorchのlstmを用いて2次元の時系列データの学習、評価を行いたいと考えています。
内容は0〜9までの数字を筆記したデータを学習させて、それぞれの識別するというものです。
発生している問題・エラーメッセージ
用いる時系列データが可変長であるため一番サイズが大きいものに合わせて、0埋めして学習させたところ、うまく識別させることが出来ませんでした。
解決方法としてどんなものがありますでしょうか。
試したこと
CrossEntropyLoss(ignore_index=0)
で学習することは試して見ましたがうまくいきませんでした。
あなたの回答
tips
プレビュー