問題タブ [pytorch-lightning]
For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.
257 問題
0
投票する
4
に答える
6573
参照
python - pytorch雷モデルの出力予測
これは潜在的に非常に簡単な質問です。PyTorch ライトニングを始めたばかりで、トレーニング後にモデルの出力を受け取る方法がわかりません。
y_train と y_test の両方の予測をある種の配列 (後のステップで PyTorch tensor または NumPy 配列) として使用して、異なるスクリプトを使用してラベルの横にプロットすることに興味があります。
私の雷モジュールには、次の機能があります。
特定の予測関数が必要ですか、それとも表示されない方法が既に実装されていますか?