matplotlib.png

最近在研究机器学习的时候,需要画一下某个函数的图形,看一看函数是什么样的,比如神经网络里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

  
  
  image.png

多个图像

• 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

  
  
  image.png

• 更多详情请参考官方文档
  官方文档