Tensorflow任务类型

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

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

不同任务类型的loss函数：

  • 二分类：binary_crossentropy
  • 多分类:
    - categorical_crossentropy 适用于label做了onehot训练，可以用tf.keras.utils.to_categorical做onehot
    - sparse_categorical_crossentropy 适用于label没有做onehot训练
  • 回归： mse
posted @ 2020-11-27 14:18  oaksharks  阅读(108)  评论(0)  编辑  收藏  举报