python写一段粒子群优化算法，对一个参数h进行优化，适应度函数为两组数据间的rmse值，越小越好

这是一段粒子群优化算法的示例代码：from pyswarm import pso# 定义适应度函数
def fitness(x):
return (rmse(x[0]))# 定义粒子群优化参数
lb = [0] # 下界
ub = [1] # 上界
args = (data1, data2) # 参数
swarmsize = 20 # 粒子群数
maxiter = 20 # 最大迭代次数# 运行粒子群优化
xopt, fopt = pso(fitness, lb, ub, args=args, swarmsize=swarmsize, maxiter=maxiter)