【发布时间】:2020-01-11 03:30:28
【问题描述】:
我有一条中间有扰动的曲线(在 t=9 处),这导致我的数据出现下降(t9)。我想拟合一个指数函数并添加一个约束,即两者(下降和上升)的面积相等。
见图:
我可以使用 optim 拟合曲线,但我无法弄清楚约束。这应该是这样的:
我已经尝试过 constrOptim,但我也愿意使用其他求解器。
y <-c(170, 160, 145, 127, 117, 74, 76, 78, 101, 115, 120, 70, 64, 65)
t <- seq(1,14,1)
# starting values:
lm <-lm(log(y) ~ log(t))
# Exp. Least-Squares minimization:
func <-function(pars) {
a <- pars["a"]
b <- pars["b"]
fitted <- a*exp(b*t)
sum((y-fitted)^2)
}
a <-lm$coefficients[[1]]
b <-lm$coefficients[[2]]
c <-
result <- optim(c(a=a, b=b), func)
# final parameters:
a <- result$par["a"]
b <- result$par["b"]
# predict values:
pred <- a*exp(b*t)
dat = data.frame(y=y, t=t, pred=pred)
library(ggplot2)
ggplot(dat, aes(x=t, y=y)) +
geom_line() +
geom_line(data=dat, aes(x=t, y=pred), color='blue')
编辑:
我知道我需要将约束添加到上面的优化中。像这样:
i = 6:12
result <- optim(c(a=a, b=b), func, sum(y[i]-a*exp(b*t[i])=0)
但这似乎不起作用。 optim 函数不允许这种约束。
【问题讨论】:
标签: r optimization constraints