激活函数

激活函数

内建激活函数

tf.nn.sigmoid

将实数压缩到 0 到 1 之间,一般只在二分类的最后输出层使用。主要缺陷为存在梯度消失问题,计算复杂度高,输出不以 0 为中心。

sigmod

tf.nn.softmax

sigmoid 的多分类扩展,一般只在多分类问题的最后输出层使用。

softmax

tf.nn.tanh

将实数压缩到-1 到 1 之间,输出期望为 0。主要缺陷为存在梯度消失问题,计算复杂度高。

tanh

tf.nn.relu

修正线性单元,最流行的激活函数。一般隐藏层使用。主要缺陷是:输出不以 0 为中心,输入小于 0 时存在梯度消失问题(死亡 relu)。

relu

tf.nn.leaky_relu

对修正线性单元的改进,解决了死亡 relu 问题。

leaky_relu

tf.nn.elu

指数线性单元。对 relu 的改进,能够缓解死亡 relu 问题。

elu

tf.nn.selu

扩展型指数线性单元。在权重用 tf.keras.initializers.lecun_normal 初始化前提下能够对神经网络进行自归一化。不可能出现梯度爆炸或者梯度消失问题。需要和 Dropout 的变种 AlphaDropout 一起使用。

selu

tf.nn.swish

自门控激活函数。谷歌出品,相关研究指出用 swish 替代 relu 将获得轻微效果提升。

swish

模型中设置激活函数

在 keras 模型中使用激活函数一般有两种方式,一种是作为某些层的 activation 参数指定,另一种是显式添加 layers.Activation 激活层。

import numpy as np
import pandas as pd
import tensorflow as tf
from tensorflow.keras import layers,models

tf.keras.backend.clear_session()

model = models.Sequential()
model.add(layers.Dense(32,input_shape = (None,16),activation = tf.nn.relu)) #通过activation参数指定
model.add(layers.Dense(10))
model.add(layers.Activation(tf.nn.softmax))  # 显式添加layers.Activation激活层
model.summary()
上一页
下一页