Tensorflow任務類型

不同任務類型的激活函數:

  • 二分類:sigmoid 也就是S函數
  • 多分類: softmax
  • 迴歸: 迴歸直接輸出

不同任務類型的loss函數:

  • 二分類:binary_crossentropy
  • 多分類:
    - categorical_crossentropy 適用於label做了onehot訓練,可以用tf.keras.utils.to_categorical做onehot
    - sparse_categorical_crossentropy 適用於label沒有做onehot訓練
  • 迴歸: mse
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章