Tensorflow任务类型

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

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

不同任务类型的loss函数:

  • 二分类:binary_crossentropy
  • 多分类:
    - categorical_crossentropy 适用于label做了onehot训练,可以用tf.keras.utils.to_categorical做onehot
    - sparse_categorical_crossentropy 适用于label没有做onehot训练
  • 回归: mse
原文地址:https://www.cnblogs.com/oaks/p/14047702.html