Tensorflow 多任務學習 概念介紹

建立多任務圖 多任務的一個特點是單個tensor輸入(X)，多個輸出(Y_1,Y_2...)。因此在定義佔位符時要定義多個輸出。同樣也需要有多個損失函數用於分別計算每個任務的損失。具體代碼如下： # GRAPH CODE # ============ # 導入 Tensorflow import Tensorflow as tf # ====================== # 定義圖

>>阅读原文<<