模拟退火算法python代码_一个模拟退火算法 Python库

github地址guofei9987/scikit-opt​github.com

安装

$pip install scikit-opt

from sko.SA import SA

def demo_func(x):

x1, x2, x3 = x

return x1 ** 2 + (x2 - 0.05) ** 2 + x3 ** 2

sa = SA(func=demo_func, x0=[1, 1, 1])

x_star, y_star = sa.run()

print(x_star, y_star)

import matplotlib.pyplot as plt

import pandas as pd

plt.plot(pd.DataFrame(sa.f_list).cummin(axis=0))

plt.show()

TSP问题(旅行商问题)

作为demo，生成模拟数据(代码与遗传算法解决TSP问题一样，这里省略)

调用模拟退火算法

from sko.SA import SA_TSP

sa_tsp = SA_TSP(func=demo_func, x0=range(num_points))

best_points, best_distance = sa_tsp.run()

画出结果

fig, ax = plt.subplots(1, 1)

best_points_ = np.concatenate([best