Python 粒子群算法PSO

最新推荐文章于 2024-07-12 10:11:05 发布

cccont

最新推荐文章于 2024-07-12 10:11:05 发布

阅读量720

点赞数

分类专栏：机器学习Machine Learning 文章标签： python 机器学习

本文链接：https://blog.csdn.net/asfdsdg/article/details/119058153

版权

机器学习Machine Learning 专栏收录该内容

5 篇文章 1 订阅

订阅专栏

本文介绍了如何使用Scikit-Optimize库中的粒子群优化(PSO)算法进行函数优化。通过实例展示了如何设置参数、定义目标函数和运行PSO过程，最后展示了优化结果和最佳解。PSO在调整机器学习模型参数方面表现出良好的效果。

摘要由CSDN通过智能技术生成

安装

pip install scikit-opt

使用

from sko.PSO import PSO
import matplotlib.pyplot as plt

def demo_func(x):
    x1, x2, x3 = x
    return x1 ** 2 + (x2 - 0.05) ** 2 + x3 ** 2

pso = PSO(func=demo_func, n_dim=3, pop=40, max_iter=150, lb=[-1，-1，-1], ub=[1,1,1])
fitness = pso.run()
print('best_x is ', pso.gbest_x)
print('best_y is ', pso.gbest_y)

plt.plot(pso.gbest_y_hist)
plt.show()