【发布时间】:2017-11-02 04:30:50
【问题描述】:
我正在尝试使用 scipy.minimize 执行简单的最小化(模拟最大似然的基本示例)。由于某种原因,它只返回初始值。我做错了什么?
这是我的代码:
import numpy as np
from scipy.optimize import minimize
# Simulated likelihood function
# Arguments:
# theta: vector representing probabilities
# sims: vector representing uniform simulated data, e.g. [0.43, 0.11, 0.02, 0.97, 0.77]
# dataCounts: vector representing counts of actual data, e.g. [4, 10, 7]
def simLogLikelihood(theta, sims, dataCounts):
# Categorise sims using theta
simCounts = np.bincount(theta.cumsum().searchsorted(sims))
# Calculate probabilities using simulated data
simProbs = simCounts/simCounts.sum()
# Calculate likelihood using simulated probabilities and actual data
logLikelihood = (dataCounts*np.log(simProbs)).sum()
return -logLikelihood
# Set seed
np.random.seed(121)
# Generate 'true' data
trueTheta = np.array([0.1, 0.4, 0.5])
dataCounts = np.bincount(np.random.choice([0, 1, 2], 1000, p=trueTheta))
# Generate simulated data (random draws from [0, 1))
sims = np.random.random(1000)
# Choose theta to maximise likelihood
thetaStart = np.array([0.33, 0.33, 0.34])
bnds = ((0, 1), (0, 1), (0, 1))
cons = ({'type': 'eq', 'fun': lambda x: x.sum() - 1.0})
result = minimize(simLogLikelihood, x0=thetaStart, args=(sims, dataCounts), method='SLSQP', bounds=bnds, constraints=cons)
(bnds 中的界限反映了概率需要介于 0 和 1 之间的事实。cons 中的约束是概率的总和必须为 1。)
如果我运行这段代码,result 包含:
fun: 1094.7593617864004
jac: array([ 0., 0., 0.])
message: 'Optimization terminated successfully.'
nfev: 5
nit: 1
njev: 1
status: 0
success: True
x: array([ 0.33, 0.33, 0.34])
所以它只进行一次迭代并返回我开始使用的概率向量。但是很容易找到另一个目标较低的概率向量,例如[0.1, 0.4, 0.5]。出了什么问题?
【问题讨论】:
-
感谢 sascha 注意到我对 numpy 而不是 scipy 的错误引用。这些已得到纠正。
标签: python numpy scipy minimization