測試函數主要是用來評估優化算法特性的，這里我用python3繪制了部分測試函數的圖像。具體的測試函數可以結合維基百科來了解。想要顯示某個測試函數的圖片把代碼結尾對應的注釋去掉即可，具體代碼如下：
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
def draw_pic(X, Y, Z, z_max, title, z_min=0):
fig = plt.figure()
ax = Axes3D(fig)
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=plt.cm.hot)
# ax.contourf(X, Y, Z, zdir='z', offset=-2, cmap=plt.cm.hot)
ax.set_zlim(z_min, z_max)
ax.set_title(title)
# plt.savefig("./myProject/Algorithm/pic/%s.png" % title) # 保存圖片
plt.show()
def get_X_AND_Y(X_min, X_max, Y_min, Y_max):
X = np.arange(X_min, X_max, 0.1)
Y = np.arange(Y_min, Y_max, 0.1)
X, Y = np.meshgrid(X, Y)
return X, Y
# rastrigin測試函數
def Rastrigin(X_min = -5.52, X_max = 5.12, Y_min = -5.12, Y_max = 5.12):
A = 10
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = 2 * A + X ** 2 - A * np.cos(2 * np.pi * X) + Y ** 2 - A * np.cos(2 * np.pi * Y)
return X, Y, Z, 100, "Rastrigin function"
# Ackley測試函數
def Ackley(X_min = -5, X_max = 5, Y_min = -5, Y_max = 5):
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = -20 * np.exp(-0.2 * np.sqrt(0.5 * (X**2 + Y**2))) - \
np.exp(0.5 * (np.cos(2 * np.pi * X) + np.cos(2 * np.pi * Y))) + np.e + 20
return X, Y, Z, 15, "Ackley function"
# Sphere測試函數
def Sphere(X_min = -3, X_max = 3, Y_min = -3, Y_max = 3):
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = X**2 + Y**2
return X, Y, Z, 20, "Sphere function"
# beale測試函數
def Beale(X_min = -4.5, X_max = 4.5, Y_min = -4.5, Y_max = 4.5):
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = np.power(1.5 - X + X * Y, 2) + np.power(2.25 - X + X * (Y ** 2), 2) \
+ np.power(2.625 - X + X * (Y ** 3), 2)
return X, Y, Z, 150000, "Beale function"
# Booth測試函數
def Booth(X_min = -10, X_max = 10, Y_min = -10, Y_max = 10):
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = np.power(X + 2*Y - 7, 2) + np.power(2 * X + Y - 5, 2)
return X, Y, Z, 2500, "Booth function"
# Bukin測試函數
def Bukin(X_min = -15, X_max = -5, Y_min = -3, Y_max = 3):
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = 100 * np.sqrt(np.abs(Y - 0.01 * X**2)) + 0.01 * np.abs(X + 10)
return X, Y, Z, 200, "Bukin function"
# Three-hump camel測試函數
def three_humpCamel(X_min = -5, X_max = 5, Y_min = -5, Y_max = 5):
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = 2 * X**2 - 1.05 * X**4 + (1/6) * X**6 + X*Y + Y*2
return X, Y, Z, 2000, "three-hump camel function"
# H?lder table測試函數
def Holder_table(X_min = -10, X_max = 10, Y_min = -10, Y_max = 10):
X, Y = get_X_AND_Y(X_min, X_max, Y_min, Y_max)
Z = -np.abs(np.sin(X) * np.cos(Y) * np.exp(np.abs(1 - np.sqrt(X**2 + Y**2)/np.pi)))
return X, Y, Z, 0, "H?lder table function", -20
z_min = None
# X, Y, Z, z_max, title = Rastrigin()
# X, Y, Z, z_max, title = Ackley()
# X, Y, Z, z_max, title = Sphere()
# X, Y, Z, z_max, title = Beale()
X, Y, Z, z_max, title = Booth()
# X, Y, Z, z_max, title = Bukin()
# X, Y, Z, z_max, title = three_humpCamel()
# X, Y, Z, z_max, title, z_min = Holder_table()
draw_pic(X, Y, Z, z_max, title, z_min)
以下是上述代碼繪制的測試函數的圖像：
感覺圖像的顏色還不是很好看，等之后優化了來改
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索：z360901061
微信掃一掃加我為好友
QQ號聯系： 360901061
您的支持是博主寫作最大的動力，如果您喜歡我的文章，感覺我的文章對您有幫助，請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧，狠狠點擊下面給點支持吧，站長非常感激您！手機微信長按不能支付解決辦法：請將微信支付二維碼保存到相冊，切換到微信，然后點擊微信右上角掃一掃功能，選擇支付二維碼完成支付。
【本文對您有幫助就好】元
