如何检查Gekko模型约束方程满足度及编写通用检查函数？

Gekko没有内置的约束违反衡量变量，但可以通过解析模型的._equations集合，替换变量求解值后计算约束违反程度。以下是通用工具函数：

工具函数代码 #

def check_constraint_violations(m):
    # 构建变量名到求解值的映射字典
    var_map = {var.name: var.value[0] for var in m._variables if var.value is not None}
    
    # 遍历所有方程约束
    for eq in m._equations:
        eq_text = str(eq)
        # 替换方程中的变量名为对应求解值
        for var_name, val in var_map.items():
            eq_text = eq_text.replace(var_name, str(val))
        
        # 处理等式约束
        if '==' in eq_text:
            lhs, rhs = eq_text.split('==')
            try:
                lhs_val = eval(lhs.strip())
                rhs_val = eval(rhs.strip())
                violation = abs(lhs_val - rhs_val)
                print(f"等式约束: {eq}")
                print(f"  违反程度: {violation:.6f}\n")
            except Exception:
                print(f"无法解析约束: {eq}\n")
        # 处理<=不等式约束
        elif '<=' in eq_text:
            lhs, rhs = eq_text.split('<=')
            try:
                lhs_val = eval(lhs.strip())
                rhs_val = eval(rhs.strip())
                # 只有当左边大于右边时才存在违反，否则违反程度为0
                violation = max(0.0, lhs_val - rhs_val)
                print(f"不等式约束: {eq}")
                print(f"  违反程度: {violation:.6f}\n")
            except Exception:
                print(f"无法解析约束: {eq}\n")
        # 处理>=不等式约束
        elif '>=' in eq_text:
            lhs, rhs = eq_text.split('>=')
            try:
                lhs_val = eval(lhs.strip())
                rhs_val = eval(rhs.strip())
                # 只有当左边小于右边时才存在违反，否则违反程度为0
                violation = max(0.0, rhs_val - lhs_val)
                print(f"不等式约束: {eq}")
                print(f"  违反程度: {violation:.6f}\n")
            except Exception:
                print(f"无法解析约束: {eq}\n")

使用示例 #

在模型求解完成后调用该函数即可：

# 假设m是你的Gekko模型实例
m.solve(disp=False)
check_constraint_violations(m)

关键细节说明 #

• 变量值取var.value[0]：Gekko变量的value属性是数组类型，单个变量取第一个元素即可。
• 约束违反程度定义：
  • 等式约束：左右两边差值的绝对值，值越大违反越严重。
  • 不等式约束：仅当约束被突破时计算超出部分，满足约束时违反程度为0。
• 兼容性：该函数兼容Gekko支持的大多数运算符（加减乘除、幂运算、三角函数等），因为eval可直接解析Python兼容的表达式。

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