不同任务类型的激活函数:
- 二分类:
sigmoid
也就是S函数 - 多分类:
softmax
- 回归: 回归直接输出
不同任务类型的loss函数:
- 二分类:binary_crossentropy
- 多分类:
- categorical_crossentropy 适用于label做了onehot训练,可以用tf.keras.utils.to_categorical做onehot
- sparse_categorical_crossentropy 适用于label没有做onehot训练 - 回归: mse