最近在研究机器学习的时候,需要画一下某个函数的图形,看一看函数是什么样的,比如神经网络里sigmoid激励函数,那么python matplotlib是一个比较好用的工具。
单个图像
- code
#导入科学计算包numpy
import numpy as np
#matplotlib画图库
import matplotlib.pyplot as plt
#sigmoid 函数 np.exp是e的x次幂
def sigmoid(x):
return 1/(np.exp(-x)+1)
#设置标题
plt.title("sigmoid function")
#设置x轴名字
plt.xlabel("this is x axis",fontsize=15,color='b')
#设置y轴名字
plt.ylabel("this is y axis",fontsize=15,color='g')
#plot会绘制的方法三个参数分别表示x轴,y轴,以及线的颜色
plt.plot(x,sigmoid(x),'r')
#显示图片
plt.show()
-
截图1
多个图像
- code
import numpy as np
import matplotlib.pyplot as plt
def sigmoid(x):
return 1/(np.exp(-x)+1)
def relu(x):
return x * (x > 0)
plt.title("sigmoid and relu")
plt.xlabel("this is x axis",fontsize=15,color='b')
plt.ylabel("this is y axis",fontsize=15,color='g')
plt.plot(x,sigmoid(x),'r',x,relu(x),'g')
plt.axis([-5,5,0,2])
plt.show()
-
截图2
- 更多详情请参考官方文档
官方文档