在Lua上进行曲线拟合

我正在寻找算法,将表格XY数据拟合到高斯函数(也称钟形曲线)。 通过谷歌搜索,我可以找到一些用于Matlab的高斯拟合算法,其中有几个:

https://ccrma.stanford.edu/~jos/sasp/Fitting_Gaussian_Data.html

http://jila.colorado.edu/bec/BEC_for_everyone/matlabfitting.htm

其中一个似乎使用Matlab的“polyfit”函数来完成任务。

是否有人看到过Lua语言的现成算法(高斯或polyfit)? 如果没有,我将非常感谢一个人帮助创建/移植这样的算法,因为我的有限Lua技能可能需要一天的时间。

原文链接 https://stackoverflow.com/questions/8913258

点赞
stackoverflow用户837856
stackoverflow用户837856

你可以将方程重组为线性形式,然后使用 Paul Bourke 在 Linear Regression 描述的方法(在页面稍下方)。

如果需要,我可以为您演示重排过程。

如果您_真的_需要,我可以提供 Lua 语言的最佳拟合直线算法实现。

2012-01-18 20:19:11
stackoverflow用户1156608
stackoverflow用户1156608

这是我试图从嘈杂的测量数据中提取高斯拟合值的方法。

require 'gsl'
require 'math'

--x=x coordinates, y=y coordinates
--clip=relative clip/ignore level 0..1 (i.e 0.1 removes values below 10% of max amplitide)
--removeoffset=set to true if y data offset should be removed
function gaussianFit( x, y, clip, removeoffset )
    local xx = {}
    local yy = {}
    local yoffset=0

    if removeoffset==nil or removeoffset==false then
    else --remove y data offset
        yoffset=gsl.Vector(y):min()
    end

    local ymax=gsl.Vector(y):max()-yoffset

    --pick only data points that has y coord larger than clip level
    for i=1,#x do
        if (y[i]-yoffset) > (clip*ymax) then
            table.insert(xx, x[i])
            table.insert(yy, math.log(y[i]-yoffset))
        end
    end

    local xvect = gsl.Vector(xx)
    local yvect = gsl.Vector(yy)

    --fit to polynomial
    local poly3 = gsl.fit.poly(3) -- a third degree polynomial
    local fit = gsl.lsfit({xvect, poly3}, yvect, nil, "fmulti") -- fits xx and yy with poly3

    --convert to gauss coeffs
    local A2=fit:coeffs()[3]
    local A1=fit:coeffs()[2]
    local A0=fit:coeffs()[1]

    local sigma=math.sqrt(-1/(2*A2))
    local mu=A1*math.pow(sigma,2)
    local A=math.exp(A0+math.pow(mu,2)/(2*math.pow(sigma,2)))

    return sigma, mu, A
end

xx={1, 2, 3, 4, 5, 6, 7, 8, 9}
yy={1, 2, 4, 6, 4, 3, 2, 1, 1}
sigma,mu,A=gaussianFit(xx,yy,0.1,false)
print(sigma.." "..mu.." ".. A)
--prints 2.2829275461334 4.6387484511153 4.201115115886
2012-01-19 16:51:08