pythongurobi向文件写入双重问题

2024-06-01 19:16:41 发布

您现在位置:Python中文网/ 问答频道 /正文

使用gurobipy,我能够使用model.write()编写我的原始问题。我也知道我能用m.getAttr("Pi", m.getConstrs())计算对偶。现在我很好奇,如果不在代码中手动写出对偶模型,是否可以轻松地生成LP的对偶并将其写入文件,就像我们将原始问题写入.lp文件一样。以下是我尝试使用的一些玩具代码,请随意测试:

import gurobipy as gp
import os
from gurobipy import GRB

def write_to_file(model, filestring):
    directory = os.getcwd()
    filename = directory+filestring
    model.write(filename)


nodes = ['s', 'a', 'b', 'c', 'd', 'e','t']
orig = 's'
dest = 't'

arcs, capacity = gp.multidict({
    ('s', 'a') : 1,
    ('a', 'b') : 3,
    ('b', 'a') : 2,
    ('c', 'e') : 4,
    ('s', 'b') : 4,
    ('a', 'd') : 4,
    ('b', 'd') : 3,
    ('d', 't') : 9,
    ('s', 'c') : 6,
    ('b', 'e') : 1,
    ('e', 't') : 4
})

m = gp.Model("Rec4")

# add our variables
flow = m.addVars(arcs, name = "flow")

# cannot go over capacity
m.addConstrs(
    (flow.sum(i,j) <= capacity[i,j] for i,j in arcs), "cap")

# flow conservation
m.addConstrs(
    (flow.sum(i,'*') - flow.sum('*',i) == 0 for i in nodes if i != orig and i != dest), "conservation")

# maximize flow
obj = (flow.sum('*', dest) - flow.sum(dest, '*'))    
m.setObjective(obj, GRB.MAXIMIZE)

m.optimize()

# print solutions
if m.status == GRB.OPTIMAL:
    solution = m.getAttr('x', flow)
    for i, j in arcs:
        if solution[i, j] > 0:
            print('%s -> %s: %g' % (i, j, solution[i, j]))

write_to_file(m, "//test.lp")

########################################################################
# Dual problem
duals = m.getAttr("Pi", m.getConstrs())
print(duals)


Tags: inimportformodelflowwritedestcapacity