【发布时间】:2016-02-01 08:59:46
【问题描述】:
我想更改现有模型中的系数。目前(使用 Python API)我正在遍历约束并调用 model.chgCoeff 但它很慢。在 Python 和/或 C API 中是否有更快的方法,可能直接访问约束矩阵?
下面的示例代码。它缓慢的原因似乎主要是因为循环本身。用任何其他操作替换 chgCoeff 仍然很慢。通常我会通过使用向量运算而不是 for 循环来解决这个问题,但如果不访问约束矩阵,我认为我做不到。
from __future__ import division
import gurobipy as gp
import numpy as np
import time
N = 300
M = 2000
m = gp.Model()
m.setParam('OutputFlag', False)
masks = [np.random.rand(N) for i in range(M)]
p = 1/np.random.rand(N)
rets = [p * masks[i] - 1 for i in range(M)]
v = np.random.rand(N)*10000 * np.round(np.random.rand(N))
t = m.addVar()
x = [m.addVar(vtype=gp.GRB.SEMICONT, lb=1000, ub=v[i]) for i in range(N)]
m.update()
cons = [m.addConstr(t <= gp.LinExpr(ret, x)) for ret in rets]
m.setObjective(t, gp.GRB.MAXIMIZE)
m.update()
start_time = time.time()
m.optimize()
solve_ms = int(((time.time() - start_time)*1000))
print('First solve took %s ms' % solve_ms)
p = 1/np.random.rand(N)
rets = [p * masks[i] - 1 for i in range(M)]
start_time = time.time()
for i in range(M):
for j in range(N):
if rets[i][j] != -1:
m.chgCoeff(cons[i], x[j], -rets[i][j])
m.update()
update_ms = int(((time.time() - start_time)*1000))
print('Model update took %s ms' % update_ms)
start_time = time.time()
m.optimize()
solve_ms = int(((time.time() - start_time)*1000))
print('Second solve took %s ms' % solve_ms)
k = 2
start_time = time.time()
for i in range(M):
for j in range(N):
if rets[i][j] != -1:
k *= rets[i][j]
solve_ms = int(((time.time() - start_time)*1000))
print('Plain loop took %s ms' % solve_ms)
R = np.array(rets)
start_time = time.time()
S = np.copy(R)
copy_ms = int(((time.time() - start_time)*1000))
print('np.copy() took %s ms' % copy_ms)
输出:
First solve took 1767 ms
Model update took 2051 ms
Second solve took 1872 ms
Plain loop took 1103 ms
np.copy() took 3 ms
在大小 (2000, 300) 约束矩阵上调用 np.copy 需要 3 毫秒。我错过了整个模型更新不能那么快的根本原因吗?
【问题讨论】:
-
您能否使用一个示例模型来更新您的问题,该示例模型展示了更新操作的缓慢性?
-
@josilber 我添加了一个示例来大致展示我正在尝试做的事情。
-
你可能想试试unofficial Gurobi API,它声称更快。我从来没有检查过。从 C API 你可以change all coefficients of a row directly。
标签: python mathematical-optimization linear-programming gurobi