如何在Pyomo中每隔X次迭代获取目标函数值（使用CBC求解器时）？

可以通过配置CBC求解器输出详细迭代日志，再解析日志提取目标函数值的方式实现。以下是具体步骤：

1. 配置CBC求解器，启用迭代日志输出 #

CBC支持通过logLevel参数开启详细迭代日志，日志中会包含每次迭代的次数和对应目标函数值。你可以选择将日志输出到文件或直接捕获控制台输出：

方式1：输出日志到文件 #

from pyomo.environ import ConcreteModel, SolverFactory

# 假设已完成模型定义（变量、约束、目标函数）
model = ConcreteModel()
# ... 模型定义代码 ...

# 初始化CBC求解器并配置日志选项
solver = SolverFactory('cbc')
solver.options['logLevel'] = 2  # 开启详细迭代日志输出
log_file = 'cbc_iter_log.txt'

# 求解模型，同时将日志写入文件
results = solver.solve(model, tee=True, logfile=log_file)

方式2：直接捕获控制台日志 #

如果不想生成文件，可以通过重定向标准输出捕获日志内容：

import io
from contextlib import redirect_stdout
from pyomo.environ import ConcreteModel, SolverFactory

model = ConcreteModel()
# ... 模型定义代码 ...

solver = SolverFactory('cbc')
solver.options['logLevel'] = 2

# 捕获求解过程的日志
log_capture = io.StringIO()
with redirect_stdout(log_capture):
    results = solver.solve(model, tee=True)
log_content = log_capture.getvalue()

2. 解析日志，提取间隔X次迭代的目标函数值 #

CBC的迭代日志行格式类似Iteration: 10 Objective value: 123.45，我们可以编写解析函数筛选出每隔X次迭代的结果：

解析文件日志 #

def extract_iter_obj(log_file, interval):
    iter_obj_list = []
    with open(log_file, 'r') as f:
        for line in f:
            line = line.strip()
            if line.startswith('Iteration:'):
                parts = line.split()
                iter_num = int(parts[1])
                obj_val = float(parts[4])
                if iter_num % interval == 0:
                    iter_obj_list.append((iter_num, obj_val))
    return iter_obj_list

# 示例：每隔5次迭代提取一次目标值
interval = 5
target_values = extract_iter_obj(log_file, interval)
print(f"每隔{interval}次迭代的目标函数值：")
for iter_num, val in target_values:
    print(f"迭代{iter_num}: {val}")

解析捕获的日志内容 #

def extract_iter_obj_from_content(log_content, interval):
    iter_obj_list = []
    for line in log_content.split('\n'):
        line = line.strip()
        if line.startswith('Iteration:'):
            parts = line.split()
            iter_num = int(parts[1])
            obj_val = float(parts[4])
            if iter_num % interval == 0:
                iter_obj_list.append((iter_num, obj_val))
    return iter_obj_list

# 示例：每隔5次迭代提取一次目标值
interval = 5
target_values = extract_iter_obj_from_content(log_content, interval)
print(f"每隔{interval}次迭代的目标函数值：")
for iter_num, val in target_values:
    print(f"迭代{iter_num}: {val}")

注意事项 #

• CBC的日志格式可能因版本略有差异，若解析失败，可查看日志内容调整匹配规则
• logLevel参数值越高，日志包含的信息越多，logLevel=2已足够输出迭代相关数据
• 若求解器未输出迭代日志，可尝试将logLevel调高至3或4

内容的提问来源于stack exchange，提问作者gameveloster