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()
