我有以下优化问题:
目标函数非常简单:给定一个向量SPREAD
,我尝试找到W
要最大化的向量sum(W.SPREAD)
。
例如,在维度 3 中,这意味着我尝试最大化w1 x spread1 + w2 x spread2 + w3 x spread3
.
另外,我有三个约束c1, c2 & c3
不是在 上W
,而是在一个POS
向量 where 上POS = W2POS(W)
。
例如,在维度 3 中,约束是:
|pos1 + pos2 + pos3| < 5
|pos1| + |pos2| + |pos3| < 500
Max(pos1, pos2, pos3) < 5
我编写了下面的代码来执行一些优化,但是不遵守约束 3。尊重我的约束,我该如何解决这个问题?
我写了下面的代码:
from scipy.optimize import fmin_cobyla
import numpy as np
import pandas as pd
def W2POS(W, PRICE, BETA):
POS = (PRICE * BETA).T.dot(W)
return POS
def objective(W, SPREAD, sign = 1):
er = sum((W * SPREAD.T).sum())
return sign * er
def c1(x, *args):
""" abs(sum(c)) < 500 """
POS = W2POS(x,args[0], args[1])
return POS.apply(abs).sum()
def c2(x, *args):
""" abs(sum()) < 5 """
POS = W2POS(x,args[0], args[1])
return 5. - abs(POS.sum())
def c3(x, *args):
""" abs(max(pos)) < 5 """
POS = W2POS(x,args[0], args[1])
return 5. - POS.apply(abs).max()
# optim
W0 = np.zeros(shape=(len(BETA), 1))
sign = -1
W = fmin_cobyla(objective, W0, cons = [c1, c2, c3], args=(SPREAD,sign),
consargs=(PRICE, BETA), maxfun=100, rhobeg = 0.02).T
print 'Solution:', W
args = [PRICE, BETA]
pos = W2POS(W.T,args[0], args[1])
print 'c1 < 5:', abs(pos.sum())[0]
print 'c2 < 500:', pos.apply(abs).sum()[0]
print 'c3 < 5:', pos.apply(abs).apply(max)[0]
您可以使用一些虚拟数据来说明此代码不尊重 c3:http: //pastebin.com/gjbeePgt