2011-09-21 184 views
1

給出點的列表,我需要實現對他們的簡單線性迴歸。這部分非常簡單,代碼示例可以在很多地方找到。線性迴歸係數

我的問題是確定迴歸係數(衡量多少分合適就行了)。我如何使用Lua以編程方式確定這樣的因素?

+3

你確定它與lua有什麼關係嗎?它應該是非常普遍的,並且lua沒有內置的方法來計算它。你必須得到算法並自己編寫代碼 – LeleDumbo

+0

你是否要求執行[線性迴歸]的算法(http://en.wikipedia.org/wiki/Linear_regression)?或者你是否要求在Lua中這樣做的已經過測試的工作代碼?維基百科和谷歌有你在第一。如果你有算法,那麼試着在Lua中實現它。如果您有問題,請回過頭來問一個問題。 –

+0

@LeleDump:對,一般算法也會。 –

回答

2
e$ lua 
Lua 5.1.4 Copyright (C) 1994-2008 Lua.org, PUC-Rio 
> function calculate_MSE (points, slope, offset) 
>>  local SE = 0 
>>  local num_pts = 0 
>>  for x,y in pairs(points) do 
>>   local p = slope * x + offset 
>>   local err = y - p 
>>   SE = SE + err * err 
>>   num_pts = num_pts + 1 
>>  end 
>>  return SE/num_pts 
>> end 
> return calculate_MSE({1, 2, 3}, 1, 0)                            0> return calculate_MSE({1, 2, 3}, 1, 1)1> return calculate_MSE({1, 2, 3}, 2, 1)9.6666666666667> 
0 
> return calculate_MSE({1, 2, 3}, 1, 1) 
1 
> return calculate_MSE({1, 2, 3}, 2, 1) 
9.6666666666667 
> 
+0

謝謝,這是完美的:) –

1

不知道你的意思是什麼:迴歸的因素,但因爲這維基百科的文章上說:MSE

線性迴歸技術,如方差 估計MSE的分析,分析和使用的部分估計的MSE到 確定所研究的因子或預測因子 的統計顯着性。實驗設計的目標是以這樣的方式構建 實驗,當分析觀察時,相對於至少一個估計處理效果的大小,MSE接近於零。

它看起來像你正在尋找的因素。

你可以對MSE的維基百科文章在計算的細節。

希望我理解您的問題,它可以幫助

+0

這似乎是我在尋找的確如此。這將有助於找到一個現有算法來計算它。 –