Python绘制不同激活函数图像

pythonxuexi 2019-12-07

"""
功能:Python绘制不同激活函数图像
姓名:侯俊龙
日期:2019/12/07
"""

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-10,10)
# 绘制sigmoid图像
fig = plt.figure()
y_sigmoid = 1/(1+np.exp(-x))
ax = fig.add_subplot(221)
ax.plot(x,y_sigmoid)
ax.grid()
ax.set_title(‘(a) Sigmoid‘)

# 绘制Tanh图像
ax = fig.add_subplot(222)
y_tanh = (np.exp(x)-np.exp(-x))/(np.exp(x)+np.exp(-x))
ax.plot(x,y_tanh)
ax.grid()
ax.set_title(‘(b) Tanh‘)

# 绘制Relu图像
ax = fig.add_subplot(223)
y_relu = np.array([0*item  if item<0 else item for item in x ])
ax.plot(x,y_relu)
ax.grid()
ax.set_title(‘(c) ReLu‘)

# 绘制Leaky ReLu图像
ax = fig.add_subplot(224)
y_relu = np.array([0.2*item  if item<0 else item for item in x ])
ax.plot(x,y_relu)
ax.grid()
ax.set_title(‘(d) Leaky ReLu‘)

plt.tight_layout()
plt.show()

Python绘制不同激活函数图像

相关推荐