Tensorflow任务类型

不同任务类型的激活函数:

  • 二分类:sigmoid 也就是S函数
  • 多分类: softmax
  • 回归: 回归直接输出

不同任务类型的loss函数:

  • 二分类:binary_crossentropy
  • 多分类:
    - categorical_crossentropy 适用于label做了onehot训练,可以用tf.keras.utils.to_categorical做onehot
    - sparse_categorical_crossentropy 适用于label没有做onehot训练
  • 回归: mse
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章