matplotlib.png
最近在研究機器學(xué)習(xí)的時候,需要畫一下某個函數(shù)的圖形,看一看函數(shù)是什么樣的,比如神經(jīng)網(wǎng)絡(luò)里sigmoid激勵函數(shù),那么python matplotlib是一個比較好用的工具聊替。
單個圖像
- code
#導(dǎo)入科學(xué)計算包numpy
import numpy as np
#matplotlib畫圖庫
import matplotlib.pyplot as plt
#sigmoid 函數(shù) np.exp是e的x次冪
def sigmoid(x):
return 1/(np.exp(-x)+1)
#設(shè)置標(biāo)題
plt.title("sigmoid function")
#設(shè)置x軸名字
plt.xlabel("this is x axis",fontsize=15,color='b')
#設(shè)置y軸名字
plt.ylabel("this is y axis",fontsize=15,color='g')
#plot會繪制的方法三個參數(shù)分別表示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
- 更多詳情請參考官方文檔
官方文檔
