Python绘制不同激活函数图像

 1 """
 2 功能：Python绘制不同激活函数图像
 3 姓名：侯俊龙
 4 日期：2019/12/07
 5 """
 6 
 7 import matplotlib.pyplot as plt
 8 import numpy as np
 9 
10 x = np.linspace(-10,10)
11 # 绘制sigmoid图像
12 fig = plt.figure()
13 y_sigmoid = 1/(1+np.exp(-x))
14 ax = fig.add_subplot(221)
15 ax.plot(x,y_sigmoid)
16 ax.grid()
17 ax.set_title('(a) Sigmoid')
18 
19 # 绘制Tanh图像
20 ax = fig.add_subplot(222)
21 y_tanh = (np.exp(x)-np.exp(-x))/(np.exp(x)+np.exp(-x))
22 ax.plot(x,y_tanh)
23 ax.grid()
24 ax.set_title('(b) Tanh')
25 
26 # 绘制Relu图像
27 ax = fig.add_subplot(223)
28 y_relu = np.array([0*item  if item<0 else item for item in x ])
29 ax.plot(x,y_relu)
30 ax.grid()
31 ax.set_title('(c) ReLu')
32 
33 # 绘制Leaky ReLu图像
34 ax = fig.add_subplot(224)
35 y_relu = np.array([0.2*item  if item<0 else item for item in x ])
36 ax.plot(x,y_relu)
37 ax.grid()
38 ax.set_title('(d) Leaky ReLu')
39 
40 plt.tight_layout()
41 plt.show()

Python绘制不同激活函数图像