if name == 'main': # 实例化问题对象 problem = MyProblem() # 生成问题对象 # 快速构建算法 algorithm = ea.soea_DE_currentToBest_1_bin_templet( problem, ea.Population(Encoding='RI', NIND=20), MAXGEN=400, # 最大进化代数。 logTras=0) # 表示每隔多少代记录一次日志信息,0表示不记录。 algorithm.mutOper.F = 0.7 # 差分进化中的参数F。 algorithm.recOper.XOVR = 0.7 # 交叉概率。 # 先验知识 prophetVars = np.array([[2000, 2000, 2000, 2000, 2000, 2000]]) # 假设已知[0.4, 0.2, 0.4]为一组比较优秀的变量。 # # 求解 res = ea.optimize(algorithm, prophet=prophetVars, verbose=True, drawing=1, outputMsg=True, drawLog=True, saveFlag=True) print(res)我想把这段代码中绘图的数据保存在一个名叫date的Excel表格中,请给我完整的代码
时间: 2023-12-29 17:06:05 浏览: 75
遗传算法,背包问题_GA-Genetic-Algorithm-for-knapsack-problem.zip
下面是完整的代码,其中绘图数据会保存在名为"date.xlsx"的 Excel 表格中:
```python
import numpy as np
import openpyxl as xl
from openpyxl.chart import LineChart, Reference
from openpyxl.chart.axis import DateAxis
from geatpy import ea
from geatpy import MyProblem
if __name__ == '__main__':
# 实例化问题对象
problem = MyProblem()
# 生成问题对象
# 快速构建算法
algorithm = ea.soea_DE_currentToBest_1_bin_templet(
problem,
ea.Population(Encoding='RI', NIND=20),
MAXGEN=400, # 最大进化代数。
logTras=0
) # 表示每隔多少代记录一次日志信息,0表示不记录。
algorithm.mutOper.F = 0.7 # 差分进化中的参数F。
algorithm.recOper.XOVR = 0.7 # 交叉概率。
# 先验知识
prophetVars = np.array([[2000, 2000, 2000, 2000, 2000, 2000]])
# 假设已知[0.4, 0.2, 0.4]为一组比较优秀的变量。
# 求解
res = ea.optimize(
algorithm,
prophet=prophetVars,
verbose=True,
drawing=1,
outputMsg=True,
drawLog=True,
saveFlag=True
)
# 保存绘图数据到 Excel 表格中
wb = xl.Workbook()
ws = wb.active
ws.append(["gen", "best"])
for i in range(len(res["BestHistory"])):
ws.append([i + 1, res["BestHistory"][i]])
chart = LineChart()
chart.title = "Best Fitness"
chart.y_axis.title = "Fitness"
chart.x_axis.title = "Generation"
xvalues = Reference(ws, min_col=1, min_row=2, max_row=len(res["BestHistory"]) + 1)
yvalues = Reference(ws, min_col=2, min_row=1, max_row=len(res["BestHistory"]) + 1)
chart.add_data(yvalues, titles_from_data=True)
chart.set_categories(xvalues)
chart.x_axis = DateAxis(crossAx=500)
ws.add_chart(chart, "D2")
wb.save("date.xlsx")
print(res)
```
在运行完这段代码后,你会在当前目录下看到一个名为"date.xlsx"的 Excel 表格,里面包含了绘图数据。
阅读全文